Software

SGMCMCJax

A Python package based on JAX for stochastic gradient Monte Carlo sampling.