numpy>=1.26.2
jax>=0.4.20
jaxlib>=0.4.20
chex>=0.1.84

[docs]
ipython>=8.18.1
matplotlib>=3.8.2
myst-nb>=1.0.0
optax>=0.1.7
sphinx>=6.2.1
sphinx-autodoc-typehints>=1.23.0
sphinx-book-theme>=1.0.1

[examples]
matplotlib>=3.8.2
optax>=0.1.7

[tests]
absl-py>=2.0.0
pytest>=7.4.3
pytest-cov>=4.1.0
