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

pyroboframes

Fast ML dataloader for robot learning. LeRobot datasets, hardware video decode, multi-output formats (NumPy/MLX/PyTorch/JAX).

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

Official RoboTrace Python SDK and CLI - observability and evals for AI-powered robots (pip install robotrace-dev, import robotrace).

robometrics

RoboMetrics is a pip-installable Python library for computing robotics trajectory, prediction, temporal drift, safety, comfort, coverage, calibration, physics, and diversity metrics from NumPy arrays and simple CSV/JSON trajectory files.

robovet

Vet your robot datasets: diagnose, repair, and quality-score LeRobot-format episode data before you waste a training run.

mohou

CNN+LSTM type "classical" visuomotor behavior cloning framework

il-datasets

A package for creating datasets for IL and IRL, training agents and benchmarking

lerobot-genesis

Run, record, and evaluate robot policies in the Genesis simulator through the standard LeRobot interfaces.

dsrl

Datasets for Offline Safe Reinforcement Learning

pylasahandwritingdataset

LASA Handwriting dataset for Python3

unienv

Framework unifying robot environments and data APIs

neodroidagent

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

robohive

A Unified Framework for Robot Learning

democlean

Score robot demonstrations by motion quality. Identify jerky, inconsistent episodes before training.

rofunc

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

pyrobovision

Advanced autonomous driving perception and vision-language foundation models. Cylindrical stitching, BEV projection, Lidar fusion, SAM3/CLIP/Grounding DINO.

tf2rl

TensorFlow2 Reinforcement Learning

navground-learning

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

carla-ue5-api

Python API for communicating with the CARLA server.

lerobot-robot-ugo-pro

ugo Pro integration for LeRobot