SqueezeAttention: 2D Management of KV-Cache in LLM Inference via Layer-wise Optimal Budget - Wang, Zihao; CUI, Bin; Gan, Shaoduo (International Conference on Learning Representations, 2025)