This is very cool. I implemented a slightly more complicated version of this with class guidance a while ago: https://github.com/aestuans/mnist-diffusion

This makes me very uncomfortable.

Thank you. Yeah mobile support was an oversight from me, I added touch support now.

I wanted something that can run everywhere and WebGPU support is still behind some flags in Firefox and Safari I think, but I'm pretty excited about trying that as well.

Good point, added touch support :)