MCMC vectorization

Hello, does the MCMC vectorization mechanism is using the jax vmap or something similar? paralell seems to be pmap related. Thanks

that’s right. we use vmap under the hood

1 Like