MCMC in JAX with benchmarks: 3 ways to write a sampler | Hacker News Reader