Imitation Learning Python Packages

tianshou

An elegant PyTorch deep reinforcement learning library.

carla

Open-source simulator for autonomous driving research.

teleop

Turns your phone or VR headset into a robot arm teleoperation device by leveraging WebXR

imitation

Clean PyTorch implementations of imitation and reward learning algorithms

di-engine

OpenDILab Decision AI Engine. The Most Comprehensive Reinforcement Learning Framework B.P.

robotrace-dev

Python SDK for RoboTrace — observability and evals for AI-powered robots. Log episodes (synchronized video + sensors + actions) from rosbag2 or Hugging Face LeRobot datasets, version them, and re-roll new policy versions against historical observations.

mohou

Visuomotor imitation learning framework

il-datasets

This is a project for creating and using IL datasets based on HuggingFace weights with multithreads for performance, and benchmarking

unienv

Framework unifying robot environments and data APIs

neodroidagent

Reinforcement learning agent implementations, intended for use with the Neodroid platform

pylasahandwritingdataset

LASA Handwriting dataset for Python3

dsrl

Datasets for Offline Safe Reinforcement Learning

rofunc

🤖 The Full Process Python Package for Robot Learning from Demonstration and Robot Manipulation

hutb

人车模拟器

tf2rl

TensorFlow2 Reinforcement Learning

robohive

A Unified Framework for Robot Learning

democlean

Score robot demonstrations by motion quality

navground-learning

A Python package that interfaces navground with gymnasium to apply IL and RL to navigation.

carla-ue5-api

Open-source simulator for autonomous driving research.

magical-il

The MAGICAL benchmark suite for robust imitation learning (NeurIPS 2020)

lerobot-robot-ugo-pro

ugo Pro integration for LeRobot

carla-client-unofficial

Open-source simulator for autonomous driving research.

gymize

Unity and Python Reinforcement and Imitation Learning with Gymnasium and PettingZoo API.

maple-robotics

MAPLE (Model and Policy Learning Evaluation) - A unified CLI daemon for evaluating robotics policies across diverse simulation environments