GitHub repository: tummfm/jax-sgmc
Primary language: Python
Description: Stochastic Gradient MCMC for Jax