Startsida Utforska Hjälp
Registrera dig Logga in
david
/
flash-attention
spegling av https://github.com/Dao-AILab/flash-attention
1
0
Fork 0
Filer Ärenden 0 Wiki
Träd: d5893f3c74
Grenar Taggar

Commit History

Upphovsman SHA1 Meddelande Datum
  Tri Dao 5400fdc4ac [CE] Implement CrossEntropyLoss in Triton 1 år sedan
  Tri Dao 43ab0b5205 Mention that some CUDA extensions have only been tested on A100s 2 år sedan
  Tri Dao fa6d1ce44f Add fused_dense and dropout_add_layernorm CUDA extensions 2 år sedan