Highly customizable (and fast) recurrent RL library for JAX/MJX
-
Updated
Oct 12, 2026 - Python
Highly customizable (and fast) recurrent RL library for JAX/MJX
A simple JAX-based implementation of random search for locomotion tasks using MuJoCo XLA (MJX).
MyoSuite with MJX (GPU-accelerated) muscle-tendon support
JAX-native robotics: URDF/MJCF import, differentiable MJX dynamics, system identification, calibrated actuators & sensors, whole-body control, and per-unit attested digital twins — one calibrated model from sim to silicon.
Code for the paper "Decentralized neural dynamics and sensory constraints shape brittle star locomotion"
Run JAX and MuJoCo MJX on AMD Ryzen AI MAX (Strix Halo, gfx1151) from pip wheels - no system ROCm - with a hang-safe check that tells you which wheel set works
四足机器人 Go2 全栈控制:Convex MPC · RL(BC/PPO/域随机化)· 混合仲裁,同一套 72 试验推力协议下恢复率 94% —— 全程 CPU 训练,附与代码逐项一致的数学设计文档
Unitree H1 humanoid walking via DeepMimic-style imitation learning — MuJoCo MJX + JAX, 4096 parallel envs, 9000 FPS, full 19-DOF control against real human mocap data
Fine-tuning the Unitree G1 LSTM walking policy with MJX
Terrain locomotion + learned fall recovery (get-up) for the Unitree Go1, trained with MJX / MuJoCo Warp + Brax PPO on a single 8 GB GPU · 中文文档
Hexapod hardware, MuJoCo/MJX tripod control, and Cartesian residual reinforcement learning
Laboratório de simulação física com MuJoCo 3.15: experimentos validados contra a física analítica (queda, pêndulo, drone, carro, braço), agent skill de controle total com docs offline, pesquisa profunda auditada e benchmarks MJX/Warp em RTX 4070.
Consolidated MuJoCo and MJX quadruped reinforcement-learning experiments
Fork of MuJoCo MPC — adds miniHexa, an 18-DOF hexapod task: real-time MPC to a goal, ported to MJX for RL policy training.
Turn the MJX GR129 / F645 2.4GHz RC controller into a PC virtual Xbox gamepad with a ProMicro nRF52840. Includes the reverse engineered protocol, the sniffer firmware and a smoothing feeder.
Build physics worlds, train reinforcement learning agents, and inspect policies with MuJoCo, a 3D editor, and a 2D wind tunnel.
To associate your repository with the mjx topic, visit your repo's landing page and select "manage topics."