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