Noise prediction in Diffusion

Demonstration of diffusion with MNIST and toy data