Understanding Multi-Head Latent Attention (From DeepSeek)
shreyansh26.github.io
1 thread
A short deep-dive on Multi-Head Latent Attention (MLA) (from DeepSeek): intuition + math, then a walk from MHA → GQA → MQA → MLA, with PyTorch code and the fusion/absorption optimizations for KV-cache efficiency.