NM512/r2dreamer

Mastering Atari with Discrete World Models

Mastering Diverse Domains through World Models

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

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

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

A platform for reproducible world model research and evaluation

Official code base for LeWorldModel: Stable End-to-End Joint-Embedding Predictive Architecture from Pixels

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

A library of reinforcement learning components and agents