https://pytorch.org/blog/inside-the-matrix/
Inside the Matrix: Visualizing Matrix Multiplication, Attention and Beyond
by Team PyTorch
September 25, 2023
“Matrix multiplications (matmuls) are the building blocks of today’s ML models. This note presents mm, a visualization tool for matmuls and compositions of matmuls.”