Imitation Learning Python Packages

tianshou

An elegant PyTorch deep reinforcement learning library.

carla

Open-source simulator for autonomous driving research.

kalanos

Audit Robotics data before you train on it. Scores LeRobot, HDF5, MCAP, ROS 2, CSV and JSON recordings for timing, signal integrity and motion quality.

game-learning-runtime

Agent-first, learner-neutral runtime for interface-driven game training and evaluation.

gauntlet-robotics

Robustness evaluation harness for robot manipulation policies (VLA): perturbation sweeps, failure clustering, reproducible reports.

orbit-eval

You have a robot. orbit tells you what to do next, and how you will know it worked. Zero-dependency CLI for LeRobot datasets and evaluations: what the recording got wrong, what it never covered, whether a change survived the noise.

fieldtrial

Statistically rigorous real-world evaluation for robot policies. Plan how many rollouts you need, run them randomized and blinded from a phone at the robot, let fieldtrial switch LeRobot or openpi policies itself, and get a report that only says "better" when a pre-registered test does.

teleop

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

di-engine

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

imitation

Clean PyTorch implementations of imitation and reward learning algorithms

mohou

CNN+LSTM type "classical" visuomotor behavior cloning framework

simthinkd

A tiny decision model (265k parameters) that picks one action in about 2 ms on one CPU core, inside real-time loops.

calibra-robotics

Dataset observability and coreset selection for robotics imitation learning

mcap-data-loader

Load and process MCAP data files for machine learning and robotics training pipelines, with a LeRobot training integration.

rofunc

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

tf2rl

TensorFlow2 Reinforcement Learning

il-datasets

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

neodroidagent

Examples of agents for neodroid environments 💡

dsrl

🔥 Datasets and env wrappers for offline safe reinforcement learning

mlx-smolvla

SmolVLA on Apple Silicon: native MLX inference, LeRobot-protocol serving, and preview on-device training, with verified parity gates

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.

robohive

A unified framework for robot learning

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.

robot-report-card

Is it your data or your model? Dataset scoring + honest policy evaluation for LeRobot / SO-101