future
numpy
matplotlib
torch

[docs]
sphinx
sphinx_rtd_theme

[test]
pytest
pytest-cov
