Difference between revisions of "MuJoCo MJX"
| Line 8: | Line 8: | ||
pip install mujoco-mjx | pip install mujoco-mjx | ||
</syntaxhighlight> | </syntaxhighlight> | ||
| + | |||
| + | == Colab Tutorial == | ||
| + | |||
| + | A detailed tutorial demonstrating the use of MJX along with reinforcement learning to train humanoid and quadruped robots to locomote is available [https://colab.research.google.com/github/google-deepmind/mujoco/blob/main/mjx/tutorial.ipynb#scrollTo=MpkYHwCqk7W- here]. | ||
Revision as of 21:42, 21 May 2024
Overview
MuJoCo XLA (MJX) is a specialized extension of the MuJoCo physics engine, designed to run simulations on hardware supported by the XLA (Accelerated Linear Algebra) compiler via the JAX framework.
Installation
Install using:
pip install mujoco-mjx
Colab Tutorial
A detailed tutorial demonstrating the use of MJX along with reinforcement learning to train humanoid and quadruped robots to locomote is available here.
