shreyansh26 / FlashAttention-PyTorchLinks
Implementation of FlashAttention in PyTorch
☆168Updated 8 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 2 months ago
- DeepSeek Native Sparse Attention pytorch implementation☆95Updated last month
- ☆147Updated 6 months ago
- TransMLA: Multi-Head Latent Attention Is All You Need☆356Updated 2 weeks ago
- A collection of memory efficient attention operators implemented in the Triton language.☆278Updated last year
- [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training☆239Updated last month
- qwen-nsa☆74Updated 5 months ago
- Get down and dirty with FlashAttention2.0 in pytorch, plug in and play no complex CUDA kernels☆109Updated 2 years ago
- 青稞Talk☆145Updated last week
- 🤖FFPA: Extend FlashAttention-2 with Split-D, ~O(1) SRAM complexity for large headdim, 1.8x~3x↑🎉 vs SDPA EA.☆217Updated last month
- Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Models☆328Updated 6 months ago
- [ICLR 2025] PEARL: Parallel Speculative Decoding with Adaptive Draft Length☆111Updated 5 months ago
- flash attention tutorial written in python, triton, cuda, cutlass☆420Updated 4 months ago
- Efficient Mixture of Experts for LLM Paper List☆127Updated this week
- PyTorch bindings for CUTLASS grouped GEMM.☆146Updated 3 weeks ago
- PyTorch bindings for CUTLASS grouped GEMM.☆119Updated 3 months ago
- ☆126Updated 3 months ago
- llm theoretical performance analysis tools and support params, flops, memory and latency analysis.☆106Updated 2 months ago
- Triton implementation of Flash Attention2.0☆39Updated 2 years ago
- Triton implementation of FlashAttention2 that adds Custom Masks.☆134Updated last year
- Implement Flash Attention using Cute.☆95Updated 9 months ago
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆569Updated this week
- [ACL 2024] A novel QAT with Self-Distillation framework to enhance ultra low-bit LLMs.☆122Updated last year
- ☆42Updated last year
- (Unofficial) PyTorch implementation of grouped-query attention (GQA) from "GQA: Training Generalized Multi-Query Transformer Models from …☆179Updated last year
- ☆154Updated 2 years ago
- Tiny-DeepSpeed, a minimalistic re-implementation of the DeepSpeed library☆45Updated last month
- An easy-to-use package for implementing SmoothQuant for LLMs☆106Updated 5 months ago
- Built upon Megatron-Deepspeed and HuggingFace Trainer, EasyLLM has reorganized the code logic with a focus on usability. While enhancing …☆49Updated last year
- [ICML 2025] Official PyTorch implementation of "FlatQuant: Flatness Matters for LLM Quantization"☆162Updated 2 months ago