3 ms·FlashAttention on a Napkin:A Diagrammatic Approach to Deep Learning IO-Awareness5 points by mpweiher 2y ago