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"
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."