jax>=0.4.25
numpy
jaxopt

[dev]
pytest
chex
scipy
matplotlib
black==26.10.1
pylint
build

[doc]
sphinx>=7
pydata-sphinx-theme
sphinx-autodoc-typehints
sphinx-gallery
sphinx-design
sphinxcontrib-bibtex
nbsphinx
ipykernel
matplotlib
optax
