https://blog.ando.ai/posts/ai-flash-attn-in-a-flash/