mctx (pypi): Monte Carlo tree search in JAX.
Keywords: jax|planning|reinforcement-learning|python|machine|learning|monte-carlo-tree-search