1 related articles
nanoAlphaZero is a single-file AlphaZero implementation in JAX that trains an Elo 2700+ chess model in 24 hours on a TPU v4-32. The entire RL pipeline is one JIT-compiled JAX function.