numpy
matplotlib
tqdm
kscale
distrax
jax
mediapy
optax
scipy
xax
mujoco
mujoco-mjx
pykos

[dev]
black
darglint
mypy
pytest
ruff
