Fix ensemble mcmc multiple run

pyro-ppl / numpyro

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.

https://num.pyro.ai

Apache License 2.0

2.1k stars 227 forks source link

Closed amifalk closed 3 months ago

amifalk commented 3 months ago

Due to misordering of the code in ensemble.init, ensemble mcmc would raise an error when running more than once after initialization.

fehiepsi commented 3 months ago

Could you merge the master branch to make test pass?