← all repositories

google-deepmind/mctx

A JAX-native library implementing Monte Carlo tree search algorithms including AlphaZero and MuZero for reinforcement learning.

2.6k stars Python AgentsML Frameworks
mctx
Velocity · 7d
+1.7
★ / day
Trend
steady
star history

Mctx provides JAX-native implementations of Monte Carlo tree search algorithms such as AlphaZero, MuZero, and Gumbel MuZero. The library enables efficient batched search operations with full JIT-compilation support, leveraging hardware accelerators for high-throughput inference. It is designed to work with deep neural network parameterized environment models, making it suitable for advanced reinforcement learning research and applications.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.