jsakaya

Hi, yes, I was able to. Here's a more stripped down version - let me know if you require something smaller. ``` import numpyro import numpyro.distributions as dist from numpyro.infer...