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.
☆82Dec 31, 2025Updated 8 months ago
Alternatives and similar repositories for Stochastic-muzero
Users that are interested in Stochastic-muzero are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Pytorch Implementation of MuZero Unplugged for gym environment. This algorithm is capable of supporting a wide range of action and observ…☆36Jun 25, 2025Updated last year
- A number of agents (PPO, MuZero) with a Perceiver-based NN architecture that can be trained to achieve goals in nethack/minihack environm…☆44Sep 19, 2022Updated 3 years ago
- A clean and easy implementation of MuZero, AlphaZero and Self-Play reinforcement learning algorithms for any game.☆16Oct 15, 2024Updated last year
- A C++ pytorch implementation of MuZero☆41May 18, 2026Updated 3 months ago
- A project that provides help for using DeepMind's mctx on gym-style environments.☆66Nov 14, 2024Updated last year
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- [IEEE ToG] MiniZero: An AlphaZero and MuZero Training Framework☆142Aug 18, 2026Updated 3 weeks ago
- [NeurIPS 2023 Spotlight] LightZero: A Unified Benchmark for Monte Carlo Tree Search in General Sequential Decision Scenarios (awesome MCT…☆1,647Aug 28, 2026Updated 2 weeks ago
- Pytorch Implementation of MuZero☆356Jul 23, 2023Updated 3 years ago
- ♟️ Vectorized RL game environments in JAX☆645Mar 6, 2025Updated last year
- Using a modified version of Werner Duvaud's MuZero implementation (https://github.com/werner-duvaud/muzero-general) this reinforcement ag…☆19Jun 24, 2026Updated 2 months ago
- Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.☆944Dec 20, 2023Updated 2 years ago
- MuZero☆2,869Sep 3, 2024Updated 2 years ago
- [ICML 2024, Spotlight] EfficientZero V2: Mastering Discrete and Continuous Control with Limited Data☆124Aug 9, 2024Updated 2 years ago
- Classic MCTS example with mctx☆25May 25, 2023Updated 3 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- A clean implementation of MuZero and AlphaZero following the AlphaZero General framework. Train and Pit both algorithms against each othe…☆170Mar 28, 2021Updated 5 years ago
- Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023☆31Jul 18, 2023Updated 3 years ago
- streaming deep reinforcement learning but 4x faster with jax!☆19Jan 4, 2026Updated 8 months ago
- Reinforcement Learning (RL), allows you to develop smart, quick and self-learning systems in your business surroundings. It is an effecti…☆13Sep 25, 2019Updated 6 years ago
- Code for our TMLR paper "Distributional GFlowNets with Quantile Flows".☆13Feb 14, 2024Updated 2 years ago
- Code for Model-Free Opponent Shaping (ICML 2022)☆24Nov 18, 2022Updated 3 years ago
- Minimal JAX implementation unifying Diffusion and Flow Matching algorithms as alternative strategies for transporting data distributions.☆66Dec 19, 2025Updated 8 months ago
- Advantage Alignment Algorithms (ICLR 2025 oral)☆19Apr 7, 2025Updated last year
- Drop-in environment replacements that make your RL algorithm train faster.☆22Jun 19, 2024Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- A PyTorch implementation of DeepMind's MuZero agent☆38Dec 1, 2023Updated 2 years ago
- ☆55Apr 11, 2023Updated 3 years ago
- Official implementation for "How Should We Meta-Learn Reinforcement Learning Algorithms?"☆23Sep 7, 2025Updated last year
- Implementation of some of the Deep Distributional Reinforcement Learning Algorithms.☆26Jun 17, 2025Updated last year
- PyTorch Implementation of the Maximum a Posteriori Policy Optimisation☆84Nov 19, 2022Updated 3 years ago
- Deep reinforcement learning framework for fast prototyping based on PyTorch☆14Mar 12, 2023Updated 3 years ago
- Deep memory and sequence models in JAX☆32Jun 8, 2026Updated 3 months ago
- Implementation of MuZero with PyTorch, based on the pseudocode from DeepMind (https://arxiv.org/src/1911.08265v2/anc/pseudocode.py).☆33Aug 14, 2022Updated 4 years ago
- ☆23May 22, 2026Updated 3 months ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Cost-aware Bayesian optimization via the Pandora's box Gittins index☆13Aug 8, 2025Updated last year
- Official implementation of ICML'24 paper "Offline Multi-Objective Optimization".☆26May 24, 2026Updated 3 months ago
- OpenAI Gym environment for graph search problems such as shortest path.☆11Dec 24, 2019Updated 6 years ago
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆422Mar 18, 2026Updated 5 months ago
- Monte Carlo tree search in JAX☆2,662Updated this week
- ☆96Feb 16, 2026Updated 6 months ago
- Code for the paper "Learning to Do or Learning While Doing: Reinforcement Learning and Bayesian Optimisation for Online Continuous Tuning…☆14Nov 15, 2023Updated 2 years ago