A minimal standalone example of diffusion model. The objective is to understand the forward as well as reverse mapping process of diffusion models. The notebook contains the equations along side the code and some visualizations.
Hope you find it useful!
The forward trajectory:
Reverse trajectory: