shreyansh26 / FlashAttention-PyTorchLinks
Implementation of FlashAttention in PyTorch
☆164Updated 7 months ago
Alternatives and similar repositories for FlashAttention-PyTorch
Users that are interested in FlashAttention-PyTorch are comparing it to the libraries listed below
Sorting:
- ☆143Updated last month
- Get down and dirty with FlashAttention2.0 in pytorch, plug in and play no complex CUDA kernels☆107Updated 2 years ago
- DeepSeek Native Sparse Attention pytorch implementation☆91Updated 3 weeks ago
- Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Models☆325Updated 6 months ago
- TransMLA: Multi-Head Latent Attention Is All You Need☆343Updated last month
- ☆146Updated 5 months ago
- qwen-nsa☆74Updated 4 months ago
- [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training☆234Updated 3 weeks ago
- A collection of memory efficient attention operators implemented in the Triton language.☆278Updated last year
- ☆154Updated 2 years ago
- Efficient Mixture of Experts for LLM Paper List☆118Updated this week
- ☆42Updated last year
- 🤖FFPA: Extend FlashAttention-2 with Split-D, ~O(1) SRAM complexity for large headdim, 1.8x~3x↑🎉 vs SDPA EA.☆211Updated 3 weeks ago
- flash attention tutorial written in python, triton, cuda, cutlass☆408Updated 3 months ago
- An easy-to-use package for implementing SmoothQuant for LLMs☆105Updated 4 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆138Updated this week
- [ICLR 2025] PEARL: Parallel Speculative Decoding with Adaptive Draft Length☆108Updated 4 months ago
- 青稞Talk☆135Updated this week
- (Unofficial) PyTorch implementation of grouped-query attention (GQA) from "GQA: Training Generalized Multi-Query Transformer Models from …☆179Updated last year
- [ACL 2024] A novel QAT with Self-Distillation framework to enhance ultra low-bit LLMs.☆120Updated last year
- Implement Flash Attention using Cute.☆95Updated 8 months ago
- Triton implementation of Flash Attention2.0☆38Updated 2 years ago
- Awesome list for LLM quantization☆282Updated this week
- The CUDA version of the RWKV language model ( https://github.com/BlinkDL/RWKV-LM )☆222Updated 8 months ago
- The official implementation of the EMNLP 2023 paper LLM-FP4☆213Updated last year
- ☆123Updated 3 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆110Updated 3 months ago
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆550Updated last month
- llm theoretical performance analysis tools and support params, flops, memory and latency analysis.☆104Updated last month
- QQQ is an innovative and hardware-optimized W4A8 quantization solution for LLMs.☆138Updated last week