If anybody is interested in playing around with this on a laptop without a GPU or without otherwise installing CUDA / torch / etc. I have a site that runs a similar algorithm on GPUs in the cloud:

https://dreamscopeapp.com/editor