torch>=1.10
numpy
scikit-learn<=1.0.2
scipy

[dev]
black
twine
pre-commit

[docs]
ipython
ipykernel
sphinx<3.0.0
sphinx_rtd_theme
nbsphinx
m2r

[examples]
ipython
jupyter
matplotlib
scipy
torchvision
tqdm

[keops]
pykeops>=1.1.1

[pyro]
pyro-ppl==1.8

[test]
flake8==4.0.1
flake8-print==4.0.0
pytest
nbval
