torch>=1.3.1
torchvision>=0.6.0
numpy>=1.9.1
sklearn

[all]
requests
tqdm
bumpversion
pre-commit
recommonmark
sphinx
sphinx_rtd_theme
sphinxcontrib-katex
sphinx-autodoc-typehints
matplotlib
torchinfo
codecov
pytest

[datasets]
requests
tqdm

[dev]
bumpversion
pre-commit

[docs]
recommonmark
sphinx
sphinx_rtd_theme
sphinxcontrib-katex
sphinx-autodoc-typehints

[examples]
matplotlib
torchinfo

[tests]
codecov
pytest
