Here is a port without the visualizer:

https://twitter.com/srush_nlp/status/1719376959572980094

Here is an amazing in-browser implementation in WebGPU

https://www.answer.ai/posts/2024-09-12-gpupuzzles.html