Sampling for Numpyro breaks running in parallel show progress mode

[fehiepsi]

Could you check if Relax matching device pattern by fehiepsi · Pull Request #1101 · pyro-ppl/numpyro · GitHub resolves the issue? We will make a patch release if it fixes the issue.