annealed_vae.zip