Provably Learning a Multi-head Attention Layer.

Sitan Chen, Yuanzhi Li

Browse the full STOC paper archive.