torch>=2.5
scipy
packaging

[all]
cupy-cuda12x>=13.0
jax[cuda12]

[cupy]
cupy-cuda12x>=13.0

[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

[jax]
jax[cuda12]
