I don't understand why NUTS code is not working. bayesian hackers mail


#21

@hellocyber Recently, we build a numpy (based on JAX) backend for Pyro. I just add a test for the above model at https://github.com/pyro-ppl/numpyro/pull/131. It is very fast (100x faster than Pyro for this model) so I guess that you want to try it out. :slight_smile: