MCMC in JAX with benchmarks: 3 ways to write a sampler
jeremiecoullon.com