jax>=0.7.0
mujoco>=3.3.0
mujoco-mjx>=3.3.0
gymnasium[mujoco]>=1.2.0
imageio
einops
flax
ml_collections
casadi
numpy

[benchmark]
fire
matplotlib
pandas

[gpu]
jax[cuda12]

[test]
pytest>=8.0.0
pytest-cov
pytest-timeout
gymnasium[array-api]
