future
numpy
scipy
matplotlib
scikit-learn>=1.0
torch

[docs]
sphinx
sphinx_rtd_theme

[test]
pytest
pytest-cov
