torch>=1.4
numpy
scipy
matplotlib
corner
joblib
pytest
tqdm
multiprocess
wandb
