/jax_sgld_psgld_mala

My implementation of sgld, psgld, and mala using JAX

Primary LanguageJupyter Notebook

Stargazers