torch>=1.10
numpy
scikit-learn
scipy

[dev]
black
twine
pre-commit

[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
