Yes, I was thinking of `matmul()`, sorry about that! The visualization is everything I hoped:

https://x.com/oranlooney/status/1977728062289555967