torch>=1.11
scipy
jaxtyping>=0.2.9
typeguard~=2.13.3

[dev]
pre-commit
setuptools_scm
ufmt
twine

[docs]
myst-parser
setuptools_scm
sphinx
six
sphinx_rtd_theme
sphinx-autodoc-typehints
uncompyle6

[test]
flake8==5.0.4
flake8-print==5.0.0
pytest
