Inside the Matrix: Visualizing Matrix Multiplication, Attention and Beyond

PyTorch

An open source machine learning framework that accelerates the path from research prototyping to production deployment.