jax<0.13,>=0.4.20
jaxlib<0.13,>=0.4.20
numpy>=1.24

[dev]
pytest>=7.0
pytest-cov
ruff>=0.6.0
pre-commit>=3.0

[examples]
flax<0.13,>=0.12.9
optax>=0.1.7
