jax>=0.10.0
flax>=0.12.7

[docs]
sphinx>=7.0
sphinx-book-theme>=1.1
sphinx-design>=0.5
myst-nb>=1.1
linkify-it-py>=2.0

[test]
pytest>=7.0
optax>=0.2.8
