JaxSGMC: Modular stochastic gradient MCMC in JAX

Open in new window