brax>=0.12.1
flax>=0.8.3
jax==0.4.25
matplotlib==3.8.4
mujoco==3.2.7
mujoco-mjx==3.2.7
numpy<2
scipy>=1.12
tyro>=0.9.16
wandb
wandb-osh

[:platform_system != "Linux"]
jaxlib==0.4.25

[:platform_system == "Linux"]
jaxlib==0.4.25+cuda12.cudnn89
