scascin0 / alphazero
A working AlphaZero implementation that's simple enough to be able to understand what's going on at a quick glance, without sacrificing too much.
☆13Updated 2 years ago
Alternatives and similar repositories for alphazero:
Users that are interested in alphazero are comparing it to the libraries listed below
- JAX implementations of core Deep RL algorithms☆79Updated 2 years ago
- A number of agents (PPO, MuZero) with a Perceiver-based NN architecture that can be trained to achieve goals in nethack/minihack environm…☆40Updated 2 years ago
- General Modules for JAX☆64Updated last month
- Reinforcement Learning Assembly☆92Updated 3 years ago
- Accelerated minigrid environments with JAX☆132Updated 7 months ago
- Scaling scaling laws with board games.☆48Updated last year
- An implementation of MuZero in JAX.☆56Updated 2 years ago
- ☆50Updated last year
- Modular framework for Reinforcement Learning in python☆172Updated 2 years ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆48Updated last week
- Reinforcement learning library in JAX.☆100Updated last year
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆109Updated 7 months ago
- ☆28Updated 2 years ago
- Adapting the AlphaZero algorithm to remove the need of execution traces to train NPI.☆79Updated last year
- Web version of “Neuroevolution of Self-Interpretable Agents” (https://arxiv.org/abs/2003.08165)☆21Updated 3 years ago
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆31Updated 3 years ago
- Baselines for gymnax 🤖☆66Updated last year
- On-the-fly conversions between Jax and NumPy tensors☆49Updated 2 years ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆53Updated 2 years ago
- Tabular methods for reinforcement learning☆36Updated 4 years ago
- An implementation of AlphaZero and MCTS with neural networks for Tetris☆19Updated last week
- AlphaZero in JAX☆77Updated 11 months ago
- MultiTask Environments for Reinforcement Learning.☆74Updated 2 years ago
- Image augmentation library for Jax☆39Updated 11 months ago
- Accelerated replay buffers in JAX☆41Updated 2 years ago
- ☆74Updated 4 months ago
- Reinforcement learning in pure JAX.☆11Updated last month
- Library to compare and evaluate reward functions☆65Updated last year
- ☆100Updated last year
- A project that provides help for using DeepMind's mctx on gym-style environments.☆58Updated 4 months ago