MCTS IMPLEMENTATION IN JAX

Summary from the TLDR AI newsletter

Monte Carlo Tree Search (MCTS) is a state-of-the-art search algorithm. It was used in AlphaGo and researchers are now working to integrate it with language models. MCTS is notoriously tricky to implement.