click>=8.3.3
imageio[ffmpeg]
gymnasium<1.3
termcolor
flatten_dict
h5py
numpy
scipy
pandas
osqp
mujoco
Pillow>=10.1.0
colorednoise>=2.2.0
packaging
ml-collections>=0.1.1
myo-sim==0.2.3
scikit-learn>=1.7.2
joblib>=1.5.3
matplotlib>=3.10.8
mjviser>=0.0.14
loop-rate-limiters>=1.2.0
huggingface_hub>=0.20

[:python_version < "3.14"]
onnxruntime<1.21

[:python_version >= "3.14"]
onnxruntime>=1.24.1

[dev]
myosuite[rl]
myosuite[musclemimic]
pytest>=9.0.3
pytest-xvfb
pytest-benchmark>=4.0
black
isort
flake8

[docs]
sphinx>=7.0
sphinx-rtd-theme>=2.0
sphinx-autodoc-typehints>=2.0

[examples]

[examples:python_version >= "3.10" and python_version < "3.14"]
mink

[examples:python_version >= "3.14"]
mink>=1.3.0

[mjlab]

[mjlab:python_version >= "3.10"]
torch>=2.13.0
warp-lang>=1.12.0

[mjlab:python_version >= "3.10" and python_version < "3.14"]
mjlab>=1.4.0
mujoco-warp>=3.7.0
wandb<0.29.0,>=0.22.3

[mjx]
mujoco>=3.7
mujoco-mjx>=3.7

[mjx-cuda]
myosuite[mjx]

[mjx-cuda:python_version >= "3.10" and sys_platform != "darwin"]
jax[cuda12]<0.5.3,>=0.4.30

[mjx:python_version >= "3.10" and sys_platform != "win32"]
etils[epath]>=1.3.0
playground>=0.0.5
mediapy>=1.2.0
imageio>=2.31.0
wandb>=0.21.0

[mjx:python_version >= "3.10" and sys_platform != "win32" and sys_platform != "darwin"]
jax<0.5.3,>=0.4.20
jaxlib<0.5.3,>=0.4.20
brax>=0.10.0
flax>=0.8.0
optax>=0.1.7
orbax-checkpoint<0.11.24,>=0.11.22
ml-collections>=0.1.1

[musclemimic]
musclemimic_models==1.0.6
huggingface_hub>=0.20
scikit-learn>=1.6.1
joblib>=1.4.0

[rl]
stable-baselines3>=2.7.1
matplotlib>=3.9.0
joblib>=1.5.3
scikit-learn>=1.6.1
wandb>=0.25.0
tensorboard>=2.20.0
