Jeremie Coullon

jax-tqdm. Add a tqdm progress bar to your JAX scans and loops.

SGMCMCJax. Lightweight library of stochastic gradient MCMC algorithms written in JAX.

jax_MCMC_blog_post. Code to reproduce Jax samplers from blog post

SGMCMC_bandit_tuning. Code for paper: Stochastic Gradient MCMC with Multi-Armed Bandit Tuning: https://arxiv.org/abs/2105.13059

knowitwall. Old version of KiW. Current version is here: https://github.com/jeremiecoullon/knowitwall-django

functional_ensemble_sampler. Code for paper: https://arxiv.org/abs/2010.15181

natural_numbers_post. Code for the "Implementing natural numbers in OCaml" blog post

vae_example. Variational Autoencoder in Flax (JAX)