Skip to content

Tutorials

Welcome to the tutorials! This section walks you through the essentials for using Mujorax environments.

If you are new to Mujorax, we highly recommend working through the Essentials to get comfortable with the basics. Each tutorial builds on the other to help you gain a better understanding of each concept and become an Mujorax master in no time! ๐Ÿ˜‰

Already an Expert?

Then what are you still doing here?! ๐Ÿคจ Get out there and run some training loops! ๐Ÿš€

Prerequisites

These tutorials assume:

  • Python 3.13+ is installed with Mujorax. If not, refer to ๐Ÿ‘‰ Getting Started
  • Familiarity with Envrax [] โ€” JaxEnv, make(), and the EnvState/EnvConfig contracts
  • Basic familiarity with JAX [] โ€” particularly jax.jit, jax.vmap, and jax.random
  • Comfort with Python dataclasses [] and chex dataclasses []

If any of that is unfamiliar, we highly recommend referring to the linked resources first and come back once comfortable. You'll get more out of the tutorials that way!

Essentials

New to Mujorax?

Start here! ๐Ÿ‘‡

Each tutorial is a short, self-contained guide that includes runnable code snippets to help get you familiar with the basics.

We recommend completing the tutorials in order below to get the most out of this tutorial series.

# Tutorial Teaches
1 Using Environments How to create and use Mujorax environments
2 Configuration Tweaking the MjxPlaygroundConfig for Playground overrides, and the CPU/CUDA impl fallback
3 Rendering Capturing RGB frames with render(state) and saving rollouts as videos