sgmcmcjax (pypi): SGMCMC samplers in JAX
Keywords: bayesian-inference|jax|sampling-methods