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