h5py
numpy
scipy
sympy
einops
pandas
matplotlib
tensorboard
scikit-learn
torch>=1.10.0
tsdb>=0.2
pygrinder>=0.4

[basic]
h5py
numpy
scipy
sympy
einops
pandas
matplotlib
tensorboard
scikit-learn
torch>=1.10.0
tsdb>=0.2
pygrinder>=0.4

[dev]
black
flake8
pre-commit
jupyterlab
h5py
numpy
scipy
sympy
einops
pandas
matplotlib
tensorboard
scikit-learn
torch>=1.10.0
tsdb>=0.2
pygrinder>=0.4
torch-geometric
torch-scatter
torch-sparse
nni
pytest-xdist
pytest-cov
furo
sphinx
sphinxcontrib-bibtex
sphinxcontrib-gtagjs
sphinx-autodoc-typehints

[doc]
furo
sphinx
sphinxcontrib-bibtex
sphinxcontrib-gtagjs
sphinx-autodoc-typehints

[full]
h5py
numpy
scipy
sympy
einops
pandas
matplotlib
tensorboard
scikit-learn
torch>=1.10.0
tsdb>=0.2
pygrinder>=0.4
torch-geometric
torch-scatter
torch-sparse
nni

[optional]
torch-geometric
torch-scatter
torch-sparse
nni

[test]
pytest-xdist
pytest-cov
