Mctx

github.comtool

Monte Carlo tree search algorithms in native JAX.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   mcx github.com

   tool category #jax link ↗

   Express & compile probabilistic programs for performant inference.

2. ▲
   jaxns github.com

   tool category #jax link ↗

   Nested sampling in JAX.

   
3. ▲
   Equinox github.com

   tool category #jax link ↗

   Callable PyTrees and filtered JIT/grad transformations => neural networks in JAX.

   
4. ▲
   GPJax github.com

   tool category #jax link ↗

   Gaussian processes in JAX.

   
5. ▲
   Writing an MCMC sampler in JAX by Jeremie Coullon jeremiecoullon.com

   site category #jax link ↗

   Tutorial on the different ways to write an MCMC sampler in JAX along with speed benchmarks.

6. ▲
   JAXopt github.com

   tool category #jax link ↗

   Hardware accelerated (GPU/TPU), batchable and differentiable optimizers in JAX.

   

JAX › Libraries > New Libraries: “Monte Carlo tree search algorithms in native JAX.”