[NeurIPS 2024, spotlight] Multivariate Learned Adaptive Noise for Diffusion Models - View it on GitHub