MCMC vectorization

numpyro
1

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

2

that’s right. we use vmap under the hood

1 Like