Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.
☆925Dec 20, 2023Updated 2 years ago
Alternatives and similar repositories for EfficientZero
Users that are interested in EfficientZero are comparing it to the libraries listed below
Sorting:
- MuZero☆2,771Sep 3, 2024Updated last year
- Pytorch Implementation of MuZero☆352Jul 23, 2023Updated 2 years ago
- Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.☆869Oct 14, 2024Updated last year
- C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.☆1,272Aug 12, 2024Updated last year
- ☆360Oct 12, 2022Updated 3 years ago
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.☆866Aug 12, 2024Updated last year
- Code for "Data-Efficient Reinforcement Learning with Self-Predictive Representations"☆163Dec 21, 2021Updated 4 years ago
- Monte Carlo tree search in JAX☆2,591Sep 2, 2025Updated 5 months ago
- An implementation of MuZero in JAX.☆57Nov 8, 2022Updated 3 years ago
- [NeurIPS 2023 Spotlight] LightZero: A Unified Benchmark for Monte Carlo Tree Search in General Sequential Decision Scenarios (awesome MCT…☆1,533Feb 11, 2026Updated 2 weeks ago
- Mastering Atari with Discrete World Models☆1,009Jan 21, 2023Updated 3 years ago
- CURL: Contrastive Unsupervised Representation Learning for Sample-Efficient Reinforcement Learning☆599Oct 28, 2020Updated 5 years ago
- A clean implementation of MuZero and AlphaZero following the AlphaZero General framework. Train and Pit both algorithms against each othe…☆168Mar 28, 2021Updated 4 years ago
- A number of agents (PPO, MuZero) with a Perceiver-based NN architecture that can be trained to achieve goals in nethack/minihack environm…☆43Sep 19, 2022Updated 3 years ago
- Official codebase for Decision Transformer: Reinforcement Learning via Sequence Modeling.☆2,770Apr 29, 2024Updated last year
- Library for Model Based RL☆1,048Jul 12, 2024Updated last year
- Discovering and Achieving Goals via World Models, NeurIPS 2021☆87Jan 24, 2024Updated 2 years ago
- Mastering Diverse Domains through World Models☆2,830Sep 23, 2025Updated 5 months ago
- DrQ-v2: Improved Data-Augmented Reinforcement Learning☆431May 31, 2022Updated 3 years ago
- RL Environments in JAX 🌍☆864May 30, 2025Updated 9 months ago
- A collection of reference environments for offline reinforcement learning☆1,649Nov 18, 2024Updated last year
- Evaluating long-term memory of reinforcement learning algorithms☆164Jun 23, 2023Updated 2 years ago
- Benchmarking the Spectrum of Agent Capabilities☆519Jan 23, 2024Updated 2 years ago
- ☆1,407Dec 9, 2025Updated 2 months ago
- CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery☆83Jul 27, 2022Updated 3 years ago
- PyTorch implementation of DreamerV2 model-based RL algorithm☆237Apr 26, 2023Updated 2 years ago
- RAD: Reinforcement Learning with Augmented Data☆416Mar 29, 2021Updated 4 years ago
- High throughput synchronous and asynchronous reinforcement learning☆974Jan 29, 2026Updated last month
- [ICML 2024, Spotlight] EfficientZero V2: Mastering Discrete and Continuous Control with Limited Data☆103Aug 9, 2024Updated last year
- High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, T…☆9,172Jul 8, 2025Updated 7 months ago
- Implementation of Dreamer v3 in pytorch.☆795Feb 21, 2026Updated last week
- Reinforcement learning library in JAX.☆101Oct 22, 2023Updated 2 years ago
- ☆328Dec 19, 2024Updated last year
- Really Fast End-to-End Jax RL Implementations☆1,027Sep 9, 2024Updated last year
- bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agent☆1,529Apr 13, 2024Updated last year
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆82Mar 9, 2023Updated 2 years ago
- Docker containers of baseline agents for the Crafter environment☆30Dec 14, 2021Updated 4 years ago
- Simple and easily configurable grid world environments for reinforcement learning☆2,403Feb 19, 2026Updated last week
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆751Oct 26, 2022Updated 3 years ago