Accueil Explorer Aide
S'inscrire Connexion
david
/
flash-attention
miroir de https://github.com/Dao-AILab/flash-attention
1
0
Fork 0
Fichiers Tickets 0 Wiki
Aborescence: 6807b1ea37
Branches Tags

Historique des commits

Auteur SHA1 Message Date
  Tri Dao eb33e587e9 [LayerNorm] Rename x1 -> residual il y a 1 an
  Tri Dao 6738d9477d [LayerNorm] Implement RMS Norm il y a 1 an
  Tri Dao 5db330519a [LayerNorm] Support taking subset of input or subset of output il y a 2 ans
  Tri Dao ae137ed17a [LayerNorm] Fuse LayerScale il y a 2 ans
  Tri Dao 8c6609ae1a [LayerNorm] Support all dimensions up to 6k (if divisible by 8) il y a 2 ans
  Tri Dao fa6d1ce44f Add fused_dense and dropout_add_layernorm CUDA extensions il y a 2 ans