Learning Linear Attention in Polynomial Time - Morris Yau; Ekin Akyürek; Jiayuan Mao; Joshua B. Tenenbaum; Stefanie Jegelka; Jacob Andreas (2024)