Diffusion via BAE