torch>=2.5
scipy

[dev]
pytest
pytest-cov
black
isort
flake8

[docs]
sphinx>=7.0.0
sphinx-rtd-theme>=1.3.0
sphinx-copybutton>=0.5.0
myst-parser>=2.0.0
sphinx-autobuild>=2021.3.14
matplotlib>=3.5.0
sphinx-autodoc-typehints>=1.24.0
sphinxcontrib-bibtex>=2.5.0
sphinx-gallery>=0.13.0

[extras]
jax
cupy
