The FFN as a Key-Value Memory: Functional Specialization in Transformer Computation. - Zaryab Rahman; Fakhrud Din; Shah Khalid; Rishi Karthikeyan (Mach. Learn., 2026)