numpyro
numpyro copied to clipboard
Add IAF VAE
Now that we have an IAF autoguide, it should be relatively straightforward to add an IAF layer to our VAE example and do some comparisons.