Skip to content

Latest commit

 

History

12 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Language: English | 简体中文

MotrixLab

GitHub License Python Version Release Docs

Train robot policies in simulation, then deploy them to real hardware.

Microduck robots walking in MotrixRender after training with MotrixLab

Microduck locomotion policies trained with MotrixLab, rendered in MotrixRender — watch the HD video.

📖 Documentation: 简体中文 | English

Contents

What is MotrixLab?

MotrixLab is an open-source reinforcement learning framework for robot training, built on the high-performance MotrixSim physics engine. Define an environment once, train it with thousands of parallel environment instances using SKRL, RSL-RL, or the built-in FastSAC, and deploy the resulting policy to MuJoCo or Unitree hardware — all through a single command-line interface.

MotrixLab architecture: define an environment once, train it with SKRL, RSL-RL or FastSAC on thousands of parallel MotrixSim environments, then deploy the same policy artifact to MuJoCo or Unitree hardware

Key Features

  • Unified Interface: Provides a concise and unified reinforcement learning training and evaluation interface
  • Multi-framework Support: Supports SKRL (JAX/PyTorch), RSLRL (PyTorch), and the built-in FastSAC implementation
  • Rich Environments: Includes various robot simulation environments such as basic control, locomotion, and manipulation tasks
  • Sim-to-Real Deployment: The same policy code deploys via the deploy CLI — Sim2Sim to MuJoCo, Sim2Real to real hardware
  • High-precision, High-performance Simulation: Built on MotrixSim, a high-precision, high-performance physics engine
  • Visual Training: Supports real-time rendering and training process visualization

🚀 Quick Start

Prerequisites

Requirement Notes
Python 3.10.x The workspace pins ==3.10.*
uv Python project and dependency manager — installation guide
Git LFS Robot meshes, motion data, and videos are tracked by LFS
OS Linux x86_64 or Windows x86_64; the JAX training backend is Linux-only

1. Clone the repository

git clone https://github.com/Motphys/MotrixLab
cd MotrixLab
git lfs pull

2. Install dependencies

uv sync --all-packages

This installs all workspace packages together with PyTorch, the default training backend used by the built-in FastSAC. Third-party frameworks such as SKRL and RSLRL are optional extras.

3. Train your first policy

uv run scripts/train.py task=microduck-walk-flat/motrix.fastsac play=true

While training, the built-in dashboard shows live run progress, episode statistics, throughput, rewards, and system health:

MotrixLab training dashboard for the microduck-walk-flat fastsac task

Training runs thousands of parallel environment instances; when it finishes, the trained policy is loaded and played in the viewer automatically. Checkpoints and TensorBoard logs are saved under runs/microduck-walk-flat/; watch the curves with:

uv run tensorboard --logdir runs/microduck-walk-flat

Training finishes in minutes: mean return and episode length typically converge after about 4,000 iterations:

TensorBoard curves of a microduck-walk-flat training run: mean return and episode length converge after about 4,000 iterations

4. Replay the trained policy

Replay the latest trained policy without retraining (for example, after stopping training early with Ctrl+C):

uv run scripts/play.py env=microduck-walk-flat

A trained microduck policy replayed in the viewer:

microduck-walk.mp4

🌍 Task Environments

MotrixLab ships 50+ built-in simulation environments spanning basic control, quadruped and humanoid locomotion, whole-body motion tracking, and manipulation. The main categories:

Preview Category Example environments
go2-walk-rough Quadruped velocity tracking go2-walk-flat · go2-walk-rough · go1-walk-rough · anymalc-walk-flat
g1-walk-flat Humanoid velocity tracking g1-walk-flat · k1-walk-rough · dex-evt-walk-flat · microduck-walk-flat
g1-wbt-dance Whole-body tracking (WBT) g1-wbt-dance · k1-wbt-freekick · g1-29dof-wbt-largebox
uv run scripts/view.py env=go2-walk-rough

See the full environment gallery for all registered environments and their supported training algorithms.

🤖 Built-in Robot Models

Seven reusable robot models are registered out of the box and can be combined into any scene or task:

Screenshot Registry name Type DoF
anymal_c anymal_c Quadruped 12
dex-evt dex-evt Humanoid 23
g1-29dof g1-29dof Humanoid 29
go1 go1 Quadruped 12
go2 go2 Quadruped 12
k1 k1 Humanoid 22
microduck microduck Humanoid 14
uv run scripts/view.py robot=go2

See Supported Robots for configuration details and how to add your own model.

🏗️ What's Inside

MotrixLab is a uv workspace of nine packages:

Package PyPI name Description
motrix_deploy motrix-deploy Framework-independent artifact, backend, policy, control-loop, registry, and CLI
motrix_deploy_mujoco motrix-deploy-mujoco MuJoCo deployment backend plugin
motrix_deploy_unitree motrix-deploy-unitree Unitree SDK2 DDS hardware backend plugin
motrix_deploy_tasks motrix-deploy-tasks Concrete versioned deployment tasks and executable bootstrap
motrix_env_core motrix-env-core Environment base classes, configuration, registry, scene construction, NumPy runtime, and rendering. It contains no built-in tasks or robot assets
motrix_env_motrixsim motrix-env-motrixsim Live MotrixSim backend, renderer, and torch frontend
motrix_env_mujoco motrix-env-mujoco Compile-only MuJoCo scene backend
motrix_envs motrix-envs Built-in environments, models, data, and environment-to-deployment-profile compilers
motrix_rl motrix-rl RL-framework integration built against motrix-env-core, with SKRL, RSLRL, and FastSAC support

🤝 Contributing

Contributions are welcome! See CONTRIBUTING.md for the development environment setup, branch and commit conventions, and the configured checks (prek, ruff, dprint, mypy).

📬 Contact

Have questions or suggestions? Feel free to contact us through:

Project policies

About

A general-purpose machine learning architecture designed for robot training

Resources

Code of conduct

Contributing

Security policy

Stars

142 stars

Watchers

2 watching

Forks

Releases

Packages

Contributors

Languages