How to convert A/B testing model in pymc3 to pyro?

The modeling code looks great! It might be a bit clearer if we put obs... variables into a numpyro.plate context.