GitHub repository: siddarthsingh1/mctx
Description: Monte Carlo tree search in JAX