Official Implementation of SEA: Sparse Linear Attention with Estimated Attention Mask (ICLR 2024)
Primary LanguagePython