torch>=2.5
scipy

[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

[extras]
jax
cupy
