ホーム エクスプローラ ヘルプ
登録 サインイン
david
/
flash-attention
同期ミラー https://github.com/Dao-AILab/flash-attention
1
0
フォーク 0
ファイル 課題 0 Wiki
ブランチ: main
ブランチ タグ
flash-attention
/
flash_attn
/
ops
Neil Tenenholtz 7153673c1a Fix swiglu backwards return type (#1337) 4 週間 前
..
triton bedf877467 [CrossEntropy] Fix where labels address not aligned to 16 bytes 2 ヶ月 前
__init__.py ece539abd6 Add __init__.py files to subdirectories for installation 2 年 前
activations.py 7153673c1a Fix swiglu backwards return type (#1337) 4 週間 前
fused_dense.py 0cb595ad94 [bugfix] handle_x not define when using checkpoint_lvl = 2 (#502) 1 年間 前
layer_norm.py f1a73d0740 Run isort and black on python files 1 年間 前
rms_norm.py f1a73d0740 Run isort and black on python files 1 年間 前