jax projects
Robotics projects tagged jax.
- openpiOpen source
Open weights and code for Physical Intelligence's π0 robot foundation models.
- 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.
- OctoOpen source
An open-source generalist robot policy trained on 800k trajectories.
