Stochastic Gradient MCMC in Jax
This repository contains SGMCMC samplers for Jax.
Jax
SGMCMCJax: a lightweight JAX library for stochastic gradient Markov chain Monte Carlo algorithms
Journal of Open Source Software
This repository contains SGMCMC samplers for Jax.
Journal of Open Source Software