mcx
github.comtool
Express & compile probabilistic programs for performant inference.
- from
- JAX
- added
- 2026-10-10
- likes
- 0
similar
-
Mctx
github.com
Monte Carlo tree search algorithms in native JAX.
-
mpi4jax
github.com
Combine MPI operations with your Jax code on CPUs and GPUs.
-
Pax
github.com
A Jax-based machine learning framework for training large scale models.
-
JAXopt
github.com
Hardware accelerated (GPU/TPU), batchable and differentiable optimizers in JAX.
-
-
SymJAX
github.com
Symbolic CPU/GPU/TPU programming.
JAX › Libraries > New Libraries: “Express & compile probabilistic programs for performant inference.”