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.

mohou

CNN+LSTM type "classical" visuomotor behavior cloning framework

rofunc

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

unienv

Framework unifying robot environments and data APIs

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 💡

zpe-robotics

ZPE-Robotics V0.0: DETERMINISTIC MOTION KERNEL: Wire-V1 Transport | Replay | Search Without Decode | Anomaly Detection | 187x Real-Data

hutb

Python API for communicating with the HUTB simulator.

pylasahandwritingdataset

LASA Handwriting dataset for Python3

dsrl

🔥 Datasets and env wrappers for offline safe reinforcement learning

tf2rl

TensorFlow2 Reinforcement Learning

robohive

A unified framework for robot learning

magical-il

MAGICAL is a benchmark suite for robust imitation learning

democlean

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

navground-learning

A Python package that interfaces navground with ML-libraries to learn navigating using IL and RL.

carla-client-unofficial

Python API for communicating with the CARLA server.

lerobot-robot-ugo-pro

ugo Pro integration for LeRobot

carla-ue5-api

Open-source simulator for autonomous driving research.

pgdrive

PGDrive: an open-ended driving simulator with infinite scenes from procedural generation

gymize

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