Efficiently Vectorized MCMC on Modern Accelerators

Hugh Dance, Pierre Glaser, Peter Orbanz +1

With the advent of automatic vectorization tools (e.g., JAX's $\text{vmap}$), writing multi-chain MCMC algorithms is often now as simple as invoking those tools on single-chain c…