numpy>=2.0.0
scipy>=1.17.0
jax>=0.7.0
mujoco>=3.3.0
mujoco-mjx>=3.3.0
gymnasium[mujoco]>=1.2.0
imageio
einops
flax
ml_collections
casadi
array-api-compat
array-api-extra

[benchmark]
fire
matplotlib
pandas
pyinstrument

[gpu]
jax[cuda12]
