Analyzing and Improving the Training Dynamics of Diffusion Models (EDM2) - View it on GitHub