Orbit: A Unified Simulation Framework for Interactive Robot Learning Environments
This framework addresses the need for efficient and scalable simulation environments in robot learning research, though it is incremental as it builds on existing simulation tools like NVIDIA Isaac Sim.
The authors introduced Orbit, a unified simulation framework for robot learning that provides modular tools for creating photo-realistic environments and supports training reinforcement learning policies and collecting demonstration datasets efficiently. The framework includes 16 robotic platforms, 4 sensor modalities, 10 motion generators, and over 20 benchmark tasks.
We present Orbit, a unified and modular framework for robot learning powered by NVIDIA Isaac Sim. It offers a modular design to easily and efficiently create robotic environments with photo-realistic scenes and high-fidelity rigid and deformable body simulation. With Orbit, we provide a suite of benchmark tasks of varying difficulty -- from single-stage cabinet opening and cloth folding to multi-stage tasks such as room reorganization. To support working with diverse observations and action spaces, we include fixed-arm and mobile manipulators with different physically-based sensors and motion generators. Orbit allows training reinforcement learning policies and collecting large demonstration datasets from hand-crafted or expert solutions in a matter of minutes by leveraging GPU-based parallelization. In summary, we offer an open-sourced framework that readily comes with 16 robotic platforms, 4 sensor modalities, 10 motion generators, more than 20 benchmark tasks, and wrappers to 4 learning libraries. With this framework, we aim to support various research areas, including representation learning, reinforcement learning, imitation learning, and task and motion planning. We hope it helps establish interdisciplinary collaborations in these communities, and its modularity makes it easily extensible for more tasks and applications in the future.