Porting stan model to PyMC

r_hats were a bit lower but still not great convergence