Tutorial: Training a Neural Network to Play Tic-Tac-Toe with Reinforcement Learning in Jax

In this article we'll learn how to train a neural network to play Tic-Tac-Toe using reinforcement learning in Jax.

This article will aim to be more pedagogical, so the code we'll end up with won't be super optimized, but it will be fast enough to train a model to perfect play in about 15 seconds on a laptop.

Code from this page can be found at this Github repo as well as in a Colab notebook (although the Colab notebook runs considerably more slowly).