numpy
scipy
scikit-learn
pandas
torch
sympy
pytorch_lightning

[docs]
sphinx
sphinx-gallery
sphinx_rtd_theme
numpydoc
matplotlib

[tests]
pytest
pytest-cov
