future
numpy
scipy
matplotlib
scikit-learn
torch
datasets

[dev]
black
pylint

[docs]
sphinx
sphinx_rtd_theme

[test]
pytest
pytest-cov
