← all repositories

blackjax-devs/blackjax

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

1.1k stars Python ML Frameworks
blackjax
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.

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