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
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