torch>=2.0
typing_extensions

[docs]
sphinx<7
sphinx-rtd-theme
sphinx-click
sphinx-autodoc-typehints
sphinx_automodapi

[formatting]
black
isort

[tests]
numpy
pytest
coverage
