robomimic

[Homepage]   [Documentation]   [Study Paper]   [Study Website]   [ARISE Initiative]

Latest Updates

• [06/20/2025] v0.5.0: Diffusion Policy, multi-dataset training, language-conditioned policies, and more!
• [03/11/2025] v0.4.0: support for robosuite v1.5 and migrate robomimic datasets to HuggingFace
• [10/11/2023] v0.3.1: support for extracting, training on, and visualizing depth observations for robosuite datasets
• [07/03/2023] v0.3.0: BC-Transformer and IQL :brain:, support for DeepMind MuJoCo bindings :robot:, pre-trained image reps :eye:, wandb logging :chart_with_upwards_trend:, and more
• [05/23/2022] v0.2.1: Updated website and documentation to feature more tutorials :notebook_with_decorative_cover:
• [12/16/2021] v0.2.0: Modular observation modalities and encoders :wrench:, support for MOMART datasets :open_file_folder: [release notes] [documentation]
• [08/09/2021] v0.1.0: Initial code and paper release

Colab quickstart

Get started with a quick colab notebook demo of robomimic without installing anything locally.

robomimic is a framework for robot learning from demonstration. It offers a broad set of demonstration datasets collected on robot manipulation domains and offline learning algorithms to learn from these datasets. robomimic aims to make robot learning broadly accessible and reproducible, allowing researchers and practitioners to benchmark tasks and algorithms fairly and to develop the next generation of robot learning algorithms.

Core Features

Reproducing benchmarks

The robomimic framework also makes reproducing the results from different benchmarks and datasets easy. See the datasets page for more information on downloading datasets and reproducing experiments.

Docker

You can use the Dockerfile to easily build a containerized environment for setting up robomimic with Python 3.9, Miniconda, robosuite, and PyTorch (CPU/GPU support).

To build, run: docker build -t robomimic .

To run without GPU (CPU only), run: docker run -it robomimic

To run with GPU (if available), run: docker run --gpus all -it robomimic

Troubleshooting

Please see the troubleshooting section for common fixes, or submit an issue on our github page.

Contributing to robomimic

This project is part of the broader Advancing Robot Intelligence through Simulated Environments (ARISE) Initiative, with the aim of lowering the barriers of entry for cutting-edge research at the intersection of AI and Robotics. The project originally began development in late 2018 by researchers in the Stanford Vision and Learning Lab (SVL). Now it is actively maintained and used for robotics research projects across multiple labs. We welcome community contributions to this project. For details please check our contributing guidelines.

Citation

Please cite this paper if you use this framework in your work:

@inproceedings{robomimic2021,
  title={What Matters in Learning from Offline Human Demonstrations for Robot Manipulation},
  author={Ajay Mandlekar and Danfei Xu and Josiah Wong and Soroush Nasiriany and Chen Wang and Rohun Kulkarni and Li Fei-Fei and Silvio Savarese and Yuke Zhu and Roberto Mart\'{i}n-Mart\'{i}n},
  booktitle={Conference on Robot Learning (CoRL)},
  year={2021}
}