numpy
jax
mujoco
mujoco-mjx
playground
brax
ml_collections
pyyaml
etils
tqdm
matplotlib
mediapy
pandas
h5py
wandb
opencv-python

[dev]
pytest
black
