jax>=0.8.0
jaxtyping>=0.2.20

[dev]
pre-commit

[docs]
zensical==0.0.11
mkdocstrings-python==2.0.1

[tests]
jaxlib
pytest
