Home Esplora Aiuto
Registrati Accedi
david
/
flash-attention
mirror da https://github.com/Dao-AILab/flash-attention
1
0
Forka 0
File Problemi 0 Wiki
Ramo (Branch): varlen
Rami (Branch) Tag
flash-attention
/
flash_attn
/
ops
lancerts 22339db185 remove an unused import (#960) 6 mesi fa
..
triton 22339db185 remove an unused import (#960) 6 mesi fa
__init__.py ece539abd6 Add __init__.py files to subdirectories for installation 2 anni fa
activations.py 3557e0bb8f [MLP] Implement SwiGLU with torch jiterator 1 anno fa
fused_dense.py 0cb595ad94 [bugfix] handle_x not define when using checkpoint_lvl = 2 (#502) 1 anno fa
layer_norm.py f1a73d0740 Run isort and black on python files 1 anno fa
rms_norm.py f1a73d0740 Run isort and black on python files 1 anno fa