Illustration dessinée par RepoMasters à partir de l’identifiant du dépôt et de la couleur de son langage. Crédits
PrototypeCommune

alexzhang13/flashattention2-custom-mask

Triton implementation of FlashAttention2 that adds Custom Masks.

Propriétaire
alexzhang13 (compte personnel)
Étoiles
Étoiles : 180
Forks
Forks : 16
Langage
Python
Licence
Apache-2.0
Création
2024
Sujets
attentionattention-mechanismcuda-kernelsdeep-learningflash-attentionflash-attention-2tritontriton-lang
Rang
Prototype (100 à 249 ★) Ce que veut dire ce rang
Voir sur GitHub

Autres dépôts de alexzhang13

Les 5 dépôts de alexzhang13

D’autres dépôts Python

Tous les dépôts Python