numpy
jax
pyyaml
networkx
lxml

[all]
mkdocs-material
mkdocstrings[python]
ruff
pin
mujoco
cbfpy

[cuda12]
jax[cuda12]

[cuda13]
jax[cuda13]

[dev]
mkdocs-material
mkdocstrings[python]
ruff

[examples]
mujoco
cbfpy

[test]
pin

[tpu]
jax[tpu]
