Aller au contenu principal
← Retour aux actualités

PISA : attention en blocs creux à complexité log-linéaire

Recherche
Source:arXiv
Le paper PISA propose une stratégie de sélection Top-K en pyramide qui porte l'attention en blocs creux à une complexité O(N log N), réduisant le goulot d'étranglement quadratique de la sélection des blocs ; noyaux Triton pour l'entraînement et l'Inférence.

Recevez le prochain point IA locale

Chaque semaine, une sélection du meilleur d'iaLocale : articles, actualités et ressources sur l'IA locale, et plus encore.

Après l'inscription, vous recevez un e-mail de confirmation (double opt-in). Sans confirmation, nous n'envoyons rien. Pas de spam, promis.