Legged Gym: A Framework for Massively Parallel Reinforcement Learning of Legged Locomotion
FeaturedNikita Rudin, David Hoeller, Philipp Reist, Marco Hutter · ETH Zurich, Robotic Systems Lab · 2022
Framework
PyTorch / Isaac Gym
License
Other (NOASSERTION)
Stars
2,971
Summary
Legged Gym is an open-source training framework built on NVIDIA Isaac Gym that enables massively parallel deep RL for legged robot locomotion, achieving policy training times of minutes rather than days.
Abstract Summary
Key Points
- High-level wrapper around Isaac Gym enabling thousands of parallel legged robot environments on a single GPU.
- Built-in terrain curriculum inspired by game design for progressively harder locomotion challenges.
- Pre-configured environments for Unitree A1/Go1, ANYmal, and Cassie with tuned reward functions.
- Training times reduced from days to minutes: flat terrain in under 4 minutes on an RTX 3090.
- Sim-to-real transfer validated on physical ANYmal and A1 robots without extensive domain randomization.
Massively Parallel Reinforcement Learning
Current (on-policy) reinforcement learning algorithms are divided into two parts: data collection and policy update. The policy update, which corresponds to back-propagation for neural networks, is easily performed in parallel on the GPU. Parallelizing data collection is not as straightforward. Each step consists of policy inference, simulation, reward, and observation calculation.
Task Description
A quadruped robot must learn to walk across challenging terrain, including uneven surfaces, slopes, stairs, and obstacles, while following base-heading and linear-velocity commands. We conduct most of the simulation and real-world deployment experiments on the ANYbotics ANYmal C robot.
However, in simulation, we demonstrate the broader applicability of the approach by additionally training policies for ANYmal B, ANYmal C with an attached arm, and the Unitree A1 robots. The terrains are selected to be representative of real-world environments.
Results
In this section, we study the effects of the number of parallel robots on the final performance of the policy. In order to use the total reward as a single representative metric, we have to remove the curriculum, otherwise a more performant policy sees its task difficulty increase and consequently a decrease in the total reward.
As such, we simplify the task by reducing the maximum step size of stairs and obstacles and directly train robots on the full range of difficulties.
Sources and Demonstrations
Figures are reproduced from Rudin et al. See the full paper for experimental details and the project page, when the paper links one, for demonstrations and videos.
Related Papers
RSL-RL: A Fast and Flexible Reinforcement Learning Library for Legged Robots
Nikita Rudin, David Hoeller, Marco Hutter
RSL-RL is a lightweight, high-performance PyTorch library implementing PPO tailored for locomotion. Used as the default RL backend in Legged Gym, it enables training thousands of agents in parallel on GPU.
A Heterogeneous Architecture for Robot RL Beyond GPU-Dominant Paradigms
Yufei Jia, Zhanxiang Cao, Mingrui Yu et al.
Simulation-based RL for contemporary robot control is increasingly organized around GPU-resident simulation: physics, rollout collection, and learning are placed on a single GPU-centric execution path
SafeManip: A Property-Driven Benchmark for Temporal Safety Evaluation in Robotic Manipulation
Chengyue Huang, Khang Vo Huynh, Sebastian Elbaum et al.
Robotic manipulation is typically evaluated by task success, but successful completion does not guarantee safe execution. Many rl failures are temporal: a robot may touch a clean surface after contamination or release an object before it is fully inside an enclosure.
Berkeley Humanoid: A Low-Cost 3D-Printed Humanoid for Research
Zachary Kousik, Shanpu Fang, Rohan Thakker et al.
An open-source 3D-printed humanoid robot platform designed for reinforcement-learning based locomotion research with affordable hardware.