Difference between revisions of "MuJoCo MJX"

From Humanoid Robots Wiki
Jump to: navigation, search
 
(One intermediate revision by the same user not shown)
Line 1: Line 1:
 
== Overview ==
 
== 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.
+
'''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.  Running single threaded physics simulation on the GPU is not very efficient. The advantage with MJX is that we can run environments in parallel on a hardware accelerated device.
  
 
=== Installation ===
 
=== Installation ===
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].

Latest revision as of 04:29, 22 May 2024

Overview[edit]

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. Running single threaded physics simulation on the GPU is not very efficient. The advantage with MJX is that we can run environments in parallel on a hardware accelerated device.

Installation[edit]

Install using:

 pip install mujoco-mjx

Colab Tutorial[edit]

A detailed tutorial demonstrating the use of MJX along with reinforcement learning to train humanoid and quadruped robots to locomote is available here.