Author of the post here - happy to answer any questions.
Great job! I'm working in a similar blog post and it was fun seeing how you approached it.
I was surprised the wasm implementation is fast enough, I was even considering writing webGpu compute shaders for my solver