PrototypeCommon
alexzhang13/flashattention2-custom-mask
Triton implementation of FlashAttention2 that adds Custom Masks.
- Owner
- alexzhang13 (personal account)
- Stars
- Stars: 180
- Forks
- Forks: 16
- Language
- Python
- License
- Apache-2.0
- Created
- 2024
- Topics
- attentionattention-mechanismcuda-kernelsdeep-learningflash-attentionflash-attention-2tritontriton-lang
- Tier
- Prototype (100 to 249 ★) What this tier means
More repos by alexzhang13
All 5 repos by alexzhang13- alexzhang13/rlmStars: 5,742PythonStable
- alexzhang13/rlm-minimalStars: 870PythonBeta
- alexzhang13/videogamebenchStars: 378PythonAlpha
- alexzhang13/spec-ptcStars: 230PythonPrototype
Nearby in Python
All Python repos- binbingwu/Multi-Agent-Game-LocalizerStars: 180PythonPrototype
- Simreal-AI/future-prediction-benchStars: 180PythonPrototype
- PandaSI-Labs/pandasi-agentStars: 180PythonPrototype
- fedora-infra/bodhiStars: 180PythonPrototype
- gaphor/gaphasStars: 180PythonPrototype
- coleifer/ircStars: 180PythonPrototype