Where can I set chain_method='vectorized' in PYMC 5.15.0 using blackjax

2

You can set it in pm.sample via the nuts_sampler_kwargs keyword, as in pm.sample(nuts_sampler='blackjax', nuts_sampler_kwargs={'chain_method':'vectorized'})

1 Like