blackjax-devs/blackjax
A library of Bayesian inference samplers (NUTS, HMC, etc.) implemented in JAX for CPU and GPU computation.

Velocity · 7d
+0.5
★ / day
Trend
→steady
star history
BlackJAX provides modular, fast sampling algorithms for Bayesian inference, built as composable kernels on top of JAX. It supports Hamiltonian Monte Carlo, NUTS, and other MCMC methods, targeting users who need GPU-accelerated statistical sampling for ML research or probabilistic programming integration.