equinox
jax
jaxtyping
distreqx
jsonpickle

[docs]
mkdocs
mkdocs-material
mkdocstrings[python]

[test]
pytest
optimistix
