Blackjax

BlackJAX is a highly optimized, open-source library collection of MCMC (Markov Chain Monte Carlo) sampling algorithms compiled in Python. It leverages Google"s JAX framework to execute ultra-fast, hardware-accelerated probabilistic math transformations over complex tensor layouts.