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 theEnvState/EnvConfigcontracts - Basic familiarity with JAX [] โ particularly
jax.jit,jax.vmap, andjax.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 |