← all repositories

HumanCompatibleAI/imitation

A PyTorch library providing clean implementations of imitation and reward learning algorithms for training agents from demonstrations.

1.8k stars Python ML FrameworksDomain Apps
imitation
Velocity · 7d
+0.6
★ / day
Trend
steady
star history

This library provides research-quality implementations of imitation learning algorithms including Behavioral Cloning, DAgger, GAIL, and AIRL. It enables training policy models using expert demonstration data instead of hand-engineered reward functions. Built on PyTorch and integrated with Gymnasium, it offers both discrete and continuous action space support along with reward learning capabilities for inverse reinforcement learning tasks.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.