Attending to all attentions (a try!) (Part 1)

From Bahdanau’s original idea to FlashAttention and Multi-Head Latent Attention.