unet_vae. Model architecture for the paper: https://arxiv.org/abs/1910.02058