numpy
matplotlib
pandas
Pillow>=6.2.0
scikit-learn
scipy
torch>=1.6.0
tqdm
h5py
structlog
pytorch-lightning>=1.4.0
packaging

[documentation]
sphinx>2
m2r
sphinx_rtd_theme
asteroid-sphinx-theme
jupyter_sphinx
Pygments>=2.6.1
nbsphinx
sphinx_automodapi
sphinx-copybutton
numpydoc
docutils==0.16
recommonmark

[nlp]
transformers
datasets

[test]
pytest
pytest-pep8
pytest-cov
coverage
torch-hypothesis==0.2.0
hypothesis==4.24.0
pytest-mock
flake8
torchvision>=0.7.0

[vision]
torchvision>=0.7.0
