Reinforcement learning in pure JAX.
A clean implementation based on AlphaZero for any game in any framework + tutorial + Othello/Gobang/TicTacToe/Connect4 and more
An implementation of the AlphaZero algorithm for Gomoku (also called Gobang or Five in a Row)
Open source neural network chess engine with GPU acceleration and broad hardware support.
MuZero
[NeurIPS 2023 Spotlight] LightZero: A Unified Benchmark for Monte Carlo Tree Search in General Sequential Decision Scenarios (awesome MCTS)
Implement AlphaZero/AlphaGo Zero methods on Chinese chess.
Distributed GPU-Accelerated Framework for Evolutionary Computation. Comprehensive Library of Evolutionary Algorithms & Benchmark Problems.
♟️ Vectorized RL game environments in JAX
Neural network-based chess engine capable of natural language commentary
Easily train AlphaZero-like agents on any environment you want!
