jax>=0.7.0
equinox>=0.11
jaxtyping
optax
tqdm
distreqx>=0.0.3
parax>=0.10.2

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

[test]
pytest
jax[cpu]
