equinox
jax
jaxtyping

[docs]
mkdocs
mkdocs-macros-plugin
mkdocs-material
mkdocstrings[python]

[test]
pytest
