alexzhang13/flashattention2-custom-mask
Triton implementation of FlashAttention2 that adds Custom Masks.
PythonApache-2.0
Stargazers
- A-suozhangTsinghua University
- ahxtTexas A&M University
- alexzhang13Princeton University
- alpercanberkColumbia University
- azhang3-sc
- BenfengXuBeijing
- blacker521
- dantypas-sc
- dsaha-sc
- edogariu
- hypnopump
- itsliupengBeijing
- jiatong-yu
- Juelianqvq
- konstantinjdoblerHasso Plattner Institute
- kopol2023
- LeeDoYupRunway
- liuzhaowen1218SUFE
- lkwq007CUHK
- lsvihPeking University
- Mellak
- michaelfeilgradient.ai @Preemo-Inc
- Misha24-10Moscow
- Mooler0410
- NKNY
- SakitsMIT, EECS
- SandalotsVolcanak
- situozhangShanghai Jiao Tong University
- sustcsonglinMIT
- Uwwal
- victbr
- Wuziyi616@uoft-isl
- xgbj
- yzhangcsSoochow University
- ZetangForwardPhd Student @ SCST, Soochow University
- ZhichenzzzUniversity of Washington