A simple JAX-based implementation of random search for locomotion tasks using MuJoCo XLA (MJX).
-
Updated
Jul 18, 2024 - Python
A simple JAX-based implementation of random search for locomotion tasks using MuJoCo XLA (MJX).
Highly customizable (and fast) recurrent RL library for JAX/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"
Fork of MuJoCo MPC — adds miniHexa, an 18-DOF hexapod task: real-time MPC to a goal, ported to MJX for RL policy training.
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
Add a description, image, and links to the mjx topic page so that developers can more easily learn about it.
To associate your repository with the mjx topic, visit your repo's landing page and select "manage topics."