AdaSplash-2: Faster Differentiable Sparse Attention
AdaSplash-2 एक नवीन हिस्टोग्राम-आधारित इनिशियलाइज़ेशन और स्पैरसिटी-अवेयर (sparsity-aware) GPU कार्यान्वयन पेश करता है जो -entmax स्पार्स अटेंशन को नाटकीय रूप से तेज़ करता है, जिससे मध्यम-से-उच्च स्पैरसिटी स्तरों पर FlashAttention-2 के तुलनीय प्रशिक्षण गति सक्षम होती है जबकि लॉन्ग-कॉन्टेक्स्ट सेटिंग्स में बेहतर प्रदर्शन प्राप्त होता है।