jax projects

Robotics projects tagged jax.

MuJoCoOpen source
Fast, accurate physics engine for robotics and RL research.
CleanRLOpen source
Single-file, research-friendly deep RL implementations.
BraxOpen source
Differentiable, massively parallel physics simulation in JAX.
MuJoCo PlaygroundOpen source
GPU-accelerated environments for locomotion and manipulation in MJX.