jax
jaxlib
equinox
optax

[dev]
pytest
pytest-cov
black
ruff
pre-commit
zensical
mkdocstrings-python

[tests]
pytest
pytest-cov
