How to run AIES in parallel

numpyro
1

Hi, I want to use more devices to speed up my program, but it seems to get error if I set chain_method=“parallel” in MCMC(AIES(),)