I got Pyro to perform almost as well as Stan on a complex model under variational inference, with the upside that I could refactor it all into classes and integ

I got Pyro to perform almost as well as Stan on a complex model under variational inference, with the upside that I could refactor it all into classes and integrate with a larger Python framework, all while making use of GPUs. Could it be any better? What do you see as wonky?