Semi-amortized VAE



any pointers on how to implement semi-amortized VAE ( in pyro?



If you use a pyro.optim.MultiOptimizer you can differentiate through parameter updates as described in the paper; the rest of the algorithm seems like pretty straightforward application of existing Pyro machinery. See the custom SVI objectives tutorial for more details.

If you can ask a more specific question with code we can probably be more helpful.