torch>=1.9

[dev]
pytest
black
flake8
sphinx
sphinx-autodoc-typehints
sphinx-rtd-theme
sphinxcontrib-spelling
codecov

[test]
pytest
