danijar/dreamerv2

Implementation of R2-Dreamer.

Mastering Diverse Domains through World Models

Implementation of Danijar's latest iteration for his Dreamer line of work

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

DIAMOND (DIffusion As a Model Of eNvironment Dreams) is a reinforcement learning agent trained in a diffusion world model. NeurIPS 2024 Spotlight.

Readable single-file PyTorch implementations of deep RL algorithms: DQN, PPO, SAC, TD3, DDPG and A2C, with Gymnasium and Atari support plus W&B tracking.

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

General multi-task deep RL Agent

Implementation of Q-Transformer, Scalable Offline Reinforcement Learning via Autoregressive Q-Functions, out of Google Deepmind