1-D Toy Diffusion model written in pure Pytorch - View it on GitHub