#vq-vae