jax projects
Robotics projects tagged jax.
- MuJoCoOpen source
Fast, accurate physics engine for robotics and RL research.
- openpiOpen source
Open weights and code for Physical Intelligence's π0 robot foundation models.
- BraxOpen source
Differentiable, massively parallel physics simulation in JAX.
- MuJoCo PlaygroundOpen source
GPU-accelerated environments for locomotion and manipulation in MJX.
