torch>=1.11
scipy
jaxtyping==0.2.19
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
uncompyle6<=3.9.0

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