torch>=2.0
scipy
jaxtyping
mpmath<=1.3,>=0.19

[dev]
pre-commit
setuptools_scm
ufmt
twine

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

[test]
flake8==5.0.4
flake8-print==5.0.0
pytest
typeguard~=2.13.3
