pytorch>=1.7.0
setuptools

[docs]
mkdocs

[testing]
pytest
