lucidrains/q-transformer

An elegant PyTorch offline reinforcement learning library for researchers.

High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)

Mastering Diverse Domains through World Models

Contrib package for Stable-Baselines3 - Experimental reinforcement learning (RL) code

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.

Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.

Mastering Atari with Discrete World Models

A library of reinforcement learning components and agents