google/brax

Features

• Reinforcement Learning - Differentiable physics engine for training reinforcement learning agents.
• Reinforcement Learning Environments - Differentiable physics engine for fast reinforcement learning research.
• Physics Simulation - Massively parallel differentiable rigid body physics engine in JAX.