Super cool library about diffusion models