Pytorch Implementation of Stochastic MuZero for gym environment. This algorithm is capable of supporting a wide range of action and observation spaces, including both discrete and continuous variations.
☆77Dec 31, 2025Updated 2 months ago
Alternatives and similar repositories for Stochastic-muzero
Users that are interested in Stochastic-muzero are comparing it to the libraries listed below
Sorting:
- Pytorch Implementation of MuZero Unplugged for gym environment. This algorithm is capable of supporting a wide range of action and observ…☆35Jun 25, 2025Updated 8 months 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
- A C++ pytorch implementation of MuZero☆40May 1, 2024Updated last year
- A project that provides help for using DeepMind's mctx on gym-style environments.☆65Nov 14, 2024Updated last year
- Pytorch Implementation of MuZero for gym environment. It support any Discrete , Box and Box2D configuration for the action space and obse…☆19Jan 24, 2023Updated 3 years ago
- A clean and easy implementation of MuZero, AlphaZero and Self-Play reinforcement learning algorithms for any game.☆17Oct 15, 2024Updated last year
- [IEEE ToG] MiniZero: An AlphaZero and MuZero Training Framework☆122Feb 25, 2026Updated last week
- Pytorch Implementation of MuZero☆352Jul 23, 2023Updated 2 years ago
- [NeurIPS 2023 Spotlight] LightZero: A Unified Benchmark for Monte Carlo Tree Search in General Sequential Decision Scenarios (awesome MCT…☆1,538Feb 25, 2026Updated last week
- [ICML 2024, Spotlight] EfficientZero V2: Mastering Discrete and Continuous Control with Limited Data☆105Aug 9, 2024Updated last year
- 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
- ♟️ Vectorized RL game environments in JAX☆591Mar 6, 2025Updated last year
- Classic MCTS example with mctx☆24May 25, 2023Updated 2 years ago
- Implementation of some of the Deep Distributional Reinforcement Learning Algorithms.☆25Jun 17, 2025Updated 8 months ago
- Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.☆925Dec 20, 2023Updated 2 years ago
- Advantage Alignment Algorithms (ICLR 2025 oral)☆17Apr 7, 2025Updated 11 months ago
- Swarm learning algorithm☆11Jun 2, 2021Updated 4 years ago
- MuZero☆2,777Sep 3, 2024Updated last year
- ☆54Apr 11, 2023Updated 2 years ago
- Deep memory and sequence models in JAX☆23Jan 15, 2026Updated last month
- ☆12Apr 22, 2022Updated 3 years ago
- Implementation of MuZero with PyTorch, based on the pseudocode from DeepMind (https://arxiv.org/src/1911.08265v2/anc/pseudocode.py).☆33Aug 14, 2022Updated 3 years ago
- A high throughput, end-to-end RL library for infinite-horizon tasks.☆23Oct 22, 2025Updated 4 months ago
- Adding Dreamer-v3's implementation tricks to CleanRL's PPO☆14May 19, 2023Updated 2 years ago
- Simple Distributed Reinforcement Learning Framework（シンプルな分散強化学習フレームワーク）☆58Nov 15, 2025Updated 3 months ago
- Code for our TMLR paper "Distributional GFlowNets with Quantile Flows".☆13Feb 14, 2024Updated 2 years ago
- Deep Reinforcement Learning Framework done with PyTorch☆43Mar 12, 2025Updated 11 months ago
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆396Updated this week
- AlphaZero for continuous control tasks☆23Dec 7, 2022Updated 3 years ago
- ☆18Nov 4, 2021Updated 4 years ago
- ☆46Sep 24, 2024Updated last year
- ☆22Jan 15, 2026Updated last month
- ☆22Aug 10, 2022Updated 3 years ago
- Drop-in environment replacements that make your RL algorithm train faster.☆21Jun 19, 2024Updated last year
- ☆20May 22, 2022Updated 3 years ago
- Monte Carlo tree search in JAX, with functionality to continue search from a previous subtree☆26May 2, 2025Updated 10 months ago
- JAX implementations of various deep reinforcement learning algorithms.☆26Feb 2, 2025Updated last year
- Risk Management via Anomaly Circumvent: Mnemonic Deep Learning for Midterm Stock Prediction. KDD 2019.☆23Aug 26, 2020Updated 5 years ago
- Trains an agent with Proximal Policy Optimization (PPO) to beat Winter Run☆23May 21, 2022Updated 3 years ago