Update pyro code (#1522)
Update pyro code (#1522)

* Update pyro code

* try numpyro

Co-authored-by: Oriol (ZBook) <af72bf69da82104e3d3d691a18658247d02fe853@gmail.com>
arviz-devs/arviz,arviz-devs/arviz,arviz-devs/arviz,arviz-devs/arviz