torch>=1.4.0
pyaml>=20.4.0

[docs]
docutils==0.16.0
recommonmark
sphinx==3.2.1
sphinx-markdown-tables
sphinx-rtd-theme==0.4.3
sphinx-copybutton
sphinxext-opengraph==0.4.1

[quality]
black>=20.8b1
isort>=5.5.4
flake8>=3.8.3

[sagemaker]
sagemaker

[test]
pytest
pytest-xdist
