jax>=0.4.38
numpy>=1.26
scipy>=1.12
optax>=0.2.4

[dev]
pytest>=8
build
twine
jsonschema>=4

[notebooks]
matplotlib>=3.8
ipython
jupyter
scikit-learn

[reproduce]
matplotlib>=3.8
