torch>=0.4.1
numpy
tqdm
matplotlib
tensorboard
hydra-core
hydra-submitit-launcher
nflows
