I built a browser tool to visualize how a small LLM picks its next word and see what it's attending to [P]

I made a little thing that visualizes attention. It runs SmolLM2-135M fully in the browser (onnxruntime-web, int8, no server, works on my phone).