SGMCMCJax: a lightweight JAX library for stochastic gradient Markov chain Monte Carlo algorithms

Jeremie Coullon, Christopher Nemeth. SGMCMCJax: a lightweight JAX library for stochastic gradient Markov chain Monte Carlo algorithms. J. Open Source Software, 7(72):4113, 2022. [doi]

Abstract

Abstract is missing.