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
sphinx>=7.2.6
sphinx-autodoc-typehints>=1.25.2

[examples]
matplotlib>=3.8.2
optax>=0.1.7

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