A project that provides help for using DeepMind's mctx on gym-style environments.
☆66Nov 14, 2024Updated last year
Alternatives and similar repositories for muax
Users that are interested in muax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆55Apr 11, 2023Updated 3 years ago
- An implementation of MuZero in JAX.☆58Nov 8, 2022Updated 3 years ago
- ☆10Mar 22, 2024Updated 2 years ago
- fast + parallel AlphaZero in JAX☆112Updated this week
- [IEEE ToG] MiniZero: An AlphaZero and MuZero Training Framework☆144Aug 18, 2026Updated 2 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Pytorch Implementation of Stochastic MuZero for gym environment. This algorithm is capable of supporting a wide range of action and obser…☆83Dec 31, 2025Updated 10 months ago
- Monte Carlo tree search in JAX☆2,684Oct 5, 2026Updated last month
- Adding Dreamer-v3's implementation tricks to CleanRL's PPO☆17May 19, 2023Updated 3 years ago
- ♟️ Vectorized RL game environments in JAX☆658Mar 6, 2025Updated last year
- Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.☆945Dec 20, 2023Updated 2 years ago
- MDP and RL interface for PDDL domains via PDDL.jl + POMDPs.jl.☆16Jun 14, 2024Updated 2 years ago
- [NeurIPS 2023 Spotlight] LightZero: A Unified Benchmark for Monte Carlo Tree Search in General Sequential Decision Scenarios (awesome MCT…☆1,660Oct 15, 2026Updated 3 weeks ago
- [ICML 2024, Spotlight] EfficientZero V2: Mastering Discrete and Continuous Control with Limited Data☆125Aug 9, 2024Updated 2 years ago
- MuZero implementation☆2,879Oct 29, 2026Updated last week
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- 🕹️ A diverse suite of scalable reinforcement learning environments in JAX☆876Oct 16, 2026Updated 2 weeks ago
- Pytorch Implementation of MuZero☆356Jul 23, 2023Updated 3 years ago
- ☆49Jul 22, 2026Updated 3 months ago
- An Agent-Environment interface with Jax transformation compatibility☆13Nov 17, 2023Updated 2 years ago
- 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
- ☆18Jan 16, 2025Updated last year
- Code for the paper "Batch size invariance for policy optimization"☆59Apr 2, 2023Updated 3 years ago
- RL Environments in JAX 🌍☆940Updated this week
- RL Environments in JAX 🌍☆19Dec 2, 2025Updated 11 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆12Jun 15, 2023Updated 3 years ago
- fast + parallel AlphaZero in PyTorch☆15Jan 21, 2024Updated 2 years ago
- OpenAi's gym environment wrapper to vectorize them with Ray☆22May 25, 2023Updated 3 years ago
- ☆12Jun 29, 2021Updated 5 years ago
- Attempting to replicate EfficientZero☆13Sep 14, 2022Updated 4 years ago
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆615Oct 27, 2026Updated last week
- MuZero for Combinatorial Action Spaces: open-source codebase for MA-Gumbel-AlphaZero, MA-Sampled-AlphaZero, MA-Gumbel-MuZero and MA-Sampl…☆24Jan 22, 2024Updated 2 years ago
- Anomalous versions of OpenAI Gym and PyBullet3 environments☆15Oct 24, 2021Updated 5 years ago
- Modular framework for Reinforcement Learning in python☆186Feb 1, 2023Updated 3 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- 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
- Python implementation of the Skeleton-Disk-Graph Roadmap planner☆16Aug 8, 2023Updated 3 years ago
- Flax Implementation of DreamerV3 on Crafter☆19Nov 29, 2025Updated 11 months ago
- Code to accompany the paper "Mismatched No More: Joint Model-Policy Optimization for Model-Based RL"☆20Oct 6, 2021Updated 5 years ago
- Docker containers of baseline agents for the Crafter environment☆30Dec 14, 2021Updated 4 years ago
- Parallel hyperparameter tuning with JAX☆40Sep 30, 2026Updated last month
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆425Mar 18, 2026Updated 7 months ago