Lightweight library of stochastic gradient MCMC algorithms written in JAX. - View it on GitHub