fast + parallel AlphaZero in JAX
☆110Dec 22, 2024Updated last year
Alternatives and similar repositories for turbozero
Users that are interested in turbozero are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- fast + parallel AlphaZero in PyTorch☆15Jan 21, 2024Updated 2 years ago
- Monte Carlo tree search in JAX, with functionality to continue search from a previous subtree☆26May 2, 2025Updated 10 months ago
- A project that provides help for using DeepMind's mctx on gym-style environments.☆65Nov 14, 2024Updated last year
- ♟️ Vectorized RL game environments in JAX☆593Mar 6, 2025Updated last year
- Monte Carlo tree search in JAX☆2,602Sep 2, 2025Updated 6 months ago
- Classic MCTS example with mctx☆24May 25, 2023Updated 2 years ago
- AlphaZero in JAX☆81Apr 3, 2024Updated last year
- 🕹️ A diverse suite of scalable reinforcement learning environments in JAX☆820Mar 9, 2026Updated 2 weeks ago
- ☆25Apr 16, 2024Updated last year
- cfrx is a collection of algorithms and tools for hardware-accelerated Counterfactual Regret Minimization (CFR) algorithms in Jax.☆37Mar 11, 2026Updated last week
- [IEEE ToG] MiniZero: An AlphaZero and MuZero Training Framework☆125Feb 25, 2026Updated last month
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆264Oct 31, 2025Updated 4 months ago
- Code for the paper "Harnessing Discrete Representations for Continual Reinforcement Learning"☆15Jun 16, 2024Updated last year
- [NeurIPS 2023 Spotlight] LightZero: A Unified Benchmark for Monte Carlo Tree Search in General Sequential Decision Scenarios (awesome MCT…☆1,550Updated this week
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆274Sep 22, 2025Updated 6 months ago
- ☆54Apr 11, 2023Updated 2 years ago
- Corax: Core RL in JAX☆41Feb 22, 2024Updated 2 years ago
- Jaxplorer is a Jax reinforcement learning (RL) framework for exploring new ideas.☆13Jul 19, 2024Updated last year
- A dataloader, but for JAX☆20May 17, 2024Updated last year
- RL Environments in JAX 🌍☆871May 30, 2025Updated 9 months ago
- Named Tensors for Legible Deep Learning in JAX☆217Nov 8, 2025Updated 4 months ago
- Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.☆926Dec 20, 2023Updated 2 years ago
- ☆16Jul 16, 2024Updated last year
- MuZero for Combinatorial Action Spaces: open-source codebase for MA-Gumbel-AlphaZero, MA-Sampled-AlphaZero, MA-Gumbel-MuZero and MA-Sampl…☆23Jan 22, 2024Updated 2 years ago
- ☆19Jan 16, 2025Updated last year
- Reinforcement learning in pure JAX.☆13Dec 24, 2025Updated 3 months ago
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Aug 8, 2022Updated 3 years ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆381Feb 10, 2026Updated last month
- Distributed Reinforcement Learning accelerated by Lightning Fabric☆425Mar 16, 2026Updated last week
- Code for the ICML 2020 publication "Information Particle Filter Tree: An Online Algorithm for POMDPs with Belief-Based Rewards on Continu…☆14Jul 3, 2020Updated 5 years ago
- ☆18Updated this week
- An implementation of the AlphaZero algorithm for adversarial games to be used with the machine learning framework of your choice☆12Aug 30, 2020Updated 5 years ago
- Simple single-file baselines for Q-Learning in pure-GPU setting☆238Nov 24, 2025Updated 4 months ago
- AlphaZero for continuous control tasks☆23Dec 7, 2022Updated 3 years ago
- A Simplified Pytorch Version of the Dreamer Algorithm☆152Jul 24, 2023Updated 2 years ago
- A collection of Matplotlib plot templates.☆25Oct 15, 2023Updated 2 years ago
- ☆10Jun 27, 2024Updated last year
- [ICML 2023] Official code for "DevFormer: A Symmetric Transformer for Context-Aware Device Placement"☆20Dec 7, 2024Updated last year
- An implementation of MuZero in JAX.☆57Nov 8, 2022Updated 3 years ago