MuJoCo MJX

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.