numpy
scipy
pandas
matplotlib
sklearn
pyyaml>=5.1
pytorch-lightning
decorator
more_itertools
simple_slurm
memory_profiler
