jax
optax
jaxlib
jaxopt
numpy
scipy>=1.11.0
h5py>2.6.0
matplotlib
pytest
toml

[DOCS]
sphinx
sphinxcontrib-bibtex
sphinx-rtd-theme
myst_nb
sphinx-autodoc-typehints
numpydoc

[TESTING]
pytest-html
coverage

[cuda11]
jax[cuda11]
optax
jaxlib
jaxopt

[cuda12]
jax[cuda12]
optax
jaxlib
jaxopt

[cuda13]
jax[cuda13]
optax
jaxlib
jaxopt
