State-of-the-art diffusion models in PyTorch and JAX
