Is there any way to speed up training?

[nessessence]

Hi, I really like your codebase of DDPM.
Have you implemented mix precision training or any kind of method that would speed up training?

Thanks