equinox
jax
jaxtyping
distreqx

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

[test]
pytest
optimistix
jaxopt
pytest-codeblocks
