PrototypeCommon
lucidrains/flash-attention-jax
Implementation of Flash Attention in Jax
- Owner
- lucidrains (personal account)
- Stars
- Stars: 230
- Forks
- Forks: 25
- Language
- Python
- License
- MIT
- Created
- 2022
- Topics
- artificial-intelligencedeep-learningattention-mechanismsjaxlong-context-attention
- Tier
- Prototype (100 to 249 ★) What this tier means
More repos by lucidrains
All 175 repos by lucidrains- lucidrains/vit-pytorchStars: 25,555PythonLTS
- lucidrains/DALLE2-pytorchStars: 11,306PythonStable
- lucidrains/denoising-diffusion-pytorchStars: 10,723PythonStable
- lucidrains/imagen-pytorchStars: 8,430PythonStable
- lucidrains/PaLM-rlhf-pytorchStars: 7,862PythonStable
- lucidrains/x-transformersStars: 5,962PythonStable
Nearby in Python
All Python repos- wonderfulearth/primitive-operation-painterStars: 230PythonPrototype
- technamooz/Panel_Technamooz_VPNStars: 230PythonPrototype
- michelle11032424-jpg/Interpark-Stars: 230PythonPrototype
- eyeseast/python-tablefuStars: 230PythonPrototype
- wardi/speedometerStars: 230PythonPrototype
- craigds/django-typed-modelsStars: 230PythonPrototype