Startseite Erkunden Hilfe
Registrieren Anmelden
david
/
flash-attention
Mirror von https://github.com/Dao-AILab/flash-attention
1
0
Fork 0
Dateien Issues 0 Wiki
Struktur: 9b6cba16c1
Branches Tags
flash-attention
/
flash_attn
/
ops
Tri Dao 8c20cfef49 [Rotary] Support qkv block layout from GQA vor 3 Monaten
..
triton 8c20cfef49 [Rotary] Support qkv block layout from GQA vor 3 Monaten
__init__.py ece539abd6 Add __init__.py files to subdirectories for installation vor 2 Jahren
activations.py 3557e0bb8f [MLP] Implement SwiGLU with torch jiterator vor 1 Jahr
fused_dense.py 0cb595ad94 [bugfix] handle_x not define when using checkpoint_lvl = 2 (#502) vor 1 Jahr
layer_norm.py f1a73d0740 Run isort and black on python files vor 1 Jahr
rms_norm.py f1a73d0740 Run isort and black on python files vor 1 Jahr