jax>=0.7.0
equinox>=0.11
jaxtyping
optax
tqdm
parax>=0.11.0

[docs]
mkdocs>=1.6
mkdocs-material>=9.5
mkdocstrings[python]>=0.26
pymdown-extensions>=10.7

[test]
pytest
jax[cpu]
