that would probably be pretty straightforward. although pyro doesn’t currently have any support for analytic kl divergences, so the simplest implementation of something like this in pyro would do without that.
that would probably be pretty straightforward. although pyro doesn’t currently have any support for analytic kl divergences, so the simplest implementation of something like this in pyro would do without that.