A library of reinforcement learning components and agents

Monte Carlo tree search in JAX

OpenSpiel is a collection of environments and algorithms for research in general reinforcement learning and search/planning in games.

Diffusion on syntax trees for program synthesis