! The code is available at https://github.com/liuliu/swift-diffusion… and I plan to port more app thing over to this repo (the repo supports CUDA and Apple, the app only supports iOS, so a little bit more work to make things supported in app also available in CUDA).