Pytorch Implementation of MuZero
☆352Jul 23, 2023Updated 2 years ago
Alternatives and similar repositories for muzero-pytorch
Users that are interested in muzero-pytorch are comparing it to the libraries listed below
Sorting:
- MuZero☆2,779Sep 3, 2024Updated last year
- Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.☆925Dec 20, 2023Updated 2 years ago
- A structured implementation of MuZero☆206Jun 4, 2022Updated 3 years ago
- A simple implementation of MuZero algorithm for connect4 game☆96Aug 11, 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 python implemenation of tabular MuZero for educational purposes☆21Dec 11, 2019Updated 6 years ago
- Code for "Data-Efficient Reinforcement Learning with Self-Predictive Representations"☆162Dec 21, 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
- Pytorch Implementation of Stochastic MuZero for gym environment. This algorithm is capable of supporting a wide range of action and obser…☆77Dec 31, 2025Updated 2 months ago
- Code for "Dream and Search to Control: Latent Space Planning for Continuous Control"☆12Jul 12, 2021Updated 4 years ago
- CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery☆84Jul 27, 2022Updated 3 years ago
- Docker containers of baseline agents for the Crafter environment☆30Dec 14, 2021Updated 4 years ago
- C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.☆1,274Aug 12, 2024Updated last year
- Library for Model Based RL☆1,054Jul 12, 2024Updated last year
- An implementation of MuZero in JAX.☆57Nov 8, 2022Updated 3 years ago
- ☆66Nov 3, 2021Updated 4 years ago
- [IEEE ToG] MiniZero: An AlphaZero and MuZero Training Framework☆124Feb 25, 2026Updated last week
- Learning Off-Policy with Online Planning [CoRL 2021 Best Paper Finalist]☆41Aug 27, 2022Updated 3 years ago
- ☆46Sep 24, 2024Updated last year
- RAD: Reinforcement Learning with Augmented Data☆416Mar 29, 2021Updated 4 years ago
- Repository for the paper "Planning to Explore via Self-Supervised World Models"☆234Feb 10, 2023Updated 3 years ago
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.☆867Aug 12, 2024Updated last year
- ☆361Oct 12, 2022Updated 3 years ago
- Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.☆870Oct 14, 2024Updated last year
- Code for the paper "When to Trust Your Model: Model-Based Policy Optimization"☆532Nov 22, 2022Updated 3 years ago
- Dream to Control: Learning Behaviors by Latent Imagination, implemented in PyTorch.☆321Jan 11, 2024Updated 2 years ago
- Mastering Atari with Discrete World Models☆1,010Jan 21, 2023Updated 3 years ago
- Deep Hierarchical Planning from Pixels☆115Dec 21, 2022Updated 3 years ago
- Offline Reinforcement Learning (aka Batch Reinforcement Learning) on Atari 2600 games☆560Jun 26, 2023Updated 2 years ago
- [NeurIPS 2023 Spotlight] LightZero: A Unified Benchmark for Monte Carlo Tree Search in General Sequential Decision Scenarios (awesome MCT…☆1,542Updated this week
- [NeurIPS 2022] Open source code for reusing prior computational work in RL.☆100Jul 5, 2023Updated 2 years ago
- Evaluating long-term memory of reinforcement learning algorithms☆164Jun 23, 2023Updated 2 years ago
- CURL: Contrastive Unsupervised Representation Learning for Sample-Efficient Reinforcement Learning☆600Oct 28, 2020Updated 5 years ago
- ☆101Feb 14, 2024Updated 2 years ago
- Monte Carlo tree search in JAX☆2,596Sep 2, 2025Updated 6 months ago
- PyTorch implementation of Stochastic Latent Actor-Critic(SLAC).☆94Jul 25, 2024Updated last year
- A repository of high-performing hierarchical reinforcement learning models and algorithms.☆336Mar 24, 2023Updated 2 years ago
- Dream to Control: Learning Behaviors by Latent Imagination☆583Sep 10, 2021Updated 4 years ago
- Reinforcement Learning in PyTorch☆2,274Jan 4, 2021Updated 5 years ago