torch>=2.2
numpy<2,>=1.24
tqdm
matplotlib
tensorboard
hydra-core
hydra-submitit-launcher
nflows
setproctitle
