Diffusion

Extending the flow matching framework to diffusion models.