ML Ops as Kernels

Map model operations such as MatMul, softmax, normalization, embeddings, and attention onto their kernel shapes, layouts, and bottlenecks.