Classic MCTS example with mctx
☆25May 25, 2023Updated 3 years ago
Alternatives and similar repositories for mctx-classic
Users that are interested in mctx-classic are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Monte Carlo tree search in JAX, with functionality to continue search from a previous subtree☆27May 2, 2025Updated last year
- ☆55Apr 11, 2023Updated 3 years ago
- A project that provides help for using DeepMind's mctx on gym-style environments.☆66Nov 14, 2024Updated last year
- An environment for learning formal mathematical reasoning from scratch☆71Aug 18, 2024Updated last year
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆278Sep 22, 2025Updated 8 months ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- cfrx is a collection of algorithms and tools for hardware-accelerated Counterfactual Regret Minimization (CFR) algorithms in Jax.☆39Mar 11, 2026Updated 3 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
- ♟️ Vectorized RL game environments in JAX☆620Mar 6, 2025Updated last year
- AlphaZero in JAX☆82Apr 3, 2024Updated 2 years ago
- Reinforcement learning in pure JAX.☆13Dec 24, 2025Updated 5 months ago
- Alpha Zero equipped with Transformer with various novel techniques for speedup in tree search☆28Nov 15, 2018Updated 7 years ago
- Distributed deep learning cluster simulation environment and RL-GNN resource management implementations.☆14Feb 1, 2023Updated 3 years ago
- ☆19Mar 1, 2023Updated 3 years ago
- An implementation of ESM2 in Equinox+JAX☆36Apr 20, 2026Updated last month
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Add a tqdm progress bar to your JAX scans and loops.☆130May 22, 2026Updated 3 weeks ago
- Adding Dreamer-v3's implementation tricks to CleanRL's PPO☆16May 19, 2023Updated 3 years ago
- Accelerated minigrid environments with JAX☆170Oct 20, 2025Updated 7 months ago
- ☆35Jan 29, 2023Updated 3 years ago
- Experimentation with Regularized Nash Dynamics on a GPU accelerated game☆55Apr 21, 2023Updated 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
- List of awesome JAX resources☆13Dec 8, 2022Updated 3 years ago
- Flax Implementation of DreamerV3 on Crafter☆18Nov 29, 2025Updated 6 months ago
- Monte Carlo tree search in JAX☆2,632Updated this week
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Network traffic in Python.☆24Mar 14, 2023Updated 3 years ago
- 🕹️ A diverse suite of scalable reinforcement learning environments in JAX☆840Jun 3, 2026Updated 2 weeks ago
- ☆14Aug 15, 2024Updated last year
- Code for NeurIPS 2022 paper Exploiting Reward Shifting in Value-Based Deep RL☆29Oct 29, 2023Updated 2 years ago
- Efficient Conway's Game of Life implemented in Python using NumPy.☆14May 1, 2024Updated 2 years ago
- Pytorch Implementation of Stochastic MuZero for gym environment. This algorithm is capable of supporting a wide range of action and obser…☆78Dec 31, 2025Updated 5 months ago
- A PyTorch implementation of PTSA-MCTS from [Accelerating Monte Carlo Tree Search with Probability Tree State Abstraction].☆16Oct 21, 2023Updated 2 years ago
- A "build to learn" Alpha Zero implementation using Gradient Boosted Decision Trees (LightGBM)☆88Mar 14, 2025Updated last year
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆23Feb 15, 2025Updated 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.
- Thompson Sampling based Monte Carlo Tree Search for MDPs and POMDPs☆15Jun 20, 2016Updated 9 years ago
- PEP 503 repository index for jax[cuda]☆21Jan 14, 2025Updated last year
- The official implementation of "Winning the L2RPN Challenge: Semi-Markov Afterstate Actor-Critic", ICLR 2021☆29Jul 10, 2022Updated 3 years ago
- Code for the paper Alpha Zero in Continuous Action Space (A0C) (https://arxiv.org/pdf/1805.09613.pdf)☆15Jan 19, 2021Updated 5 years ago
- Open source code for EigenGame.☆35May 15, 2023Updated 3 years ago
- AlphaZero for continuous control tasks☆23Dec 7, 2022Updated 3 years ago
- [IEEE ToG] MiniZero: An AlphaZero and MuZero Training Framework☆131May 9, 2026Updated last month