mm Matmul visualizations in 3D Tool Reference Notes Inside the Matrix: Visualizing matrix multiplications, Transformer Attention and Beyond (slides) Examples GPT2 attention head explorer