torch>=1.9.1
pyro-ppl>=1.7.0
numpy>=1.21.2
pandas>=1.3.3
tqdm==4.62.3
jupyter
matplotlib>=2.0.2
pandas
pillow
scikit-learn>=0.24.2
safetensors>=0.3.1
sphinx
sphinx_rtd_theme

[docs]
Sphinx
sphinx_rtd_theme
numpydoc
