Thinking in Pallas - Sharded MatMuls

Street Fighting Kernels for TPUs