mctx
Mctx provides a JAX-native implementation of Monte Carlo tree search algorithms, supporting models like AlphaZero and MuZero. It utilizes JIT-compilation for enhanced performance. The library is ideal for reinforcement learning research and offers configurable search methods. Installation is available through PyPI or GitHub, with practical examples included for ease of use.
