kyegomez / FlashAttention20Triton
Triton implementation of Flash Attention2.0
☆32Updated last year
Alternatives and similar repositories for FlashAttention20Triton:
Users that are interested in FlashAttention20Triton are comparing it to the libraries listed below
- PyTorch bindings for CUTLASS grouped GEMM.☆81Updated 5 months ago
- Odysseus: Playground of LLM Sequence Parallelism☆68Updated 10 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆120Updated 3 months ago
- ☆82Updated last month
- Boosting 4-bit inference kernels with 2:4 Sparsity☆72Updated 7 months ago
- An easy-to-use package for implementing SmoothQuant for LLMs☆96Updated 2 weeks ago
- [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving☆305Updated 9 months ago
- ☆92Updated 7 months ago
- ☆103Updated 7 months ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆209Updated 8 months ago
- [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training☆184Updated last week
- [ACL 2024] A novel QAT with Self-Distillation framework to enhance ultra low-bit LLMs.☆109Updated 11 months ago
- Decoding Attention is specially optimized for MHA, MQA, GQA and MLA using CUDA core for the decoding stage of LLM inference.☆36Updated 3 weeks ago
- Implement Flash Attention using Cute.☆76Updated 4 months ago
- ☆55Updated 2 weeks ago
- A collection of memory efficient attention operators implemented in the Triton language.☆262Updated 10 months ago
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆159Updated 9 months ago
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆94Updated last week
- ☆68Updated 3 months ago
- ☆82Updated 3 years ago
- Ouroboros: Speculative Decoding with Large Model Enhanced Drafting (EMNLP 2024 main)☆102Updated last month
- 16-fold memory access reduction with nearly no loss☆90Updated last month
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆106Updated 9 months ago
- [ICLR 2025] PEARL: Parallel Speculative Decoding with Adaptive Draft Length☆77Updated last week
- ☆131Updated last month
- ☆147Updated last year
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆111Updated last week
- [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding☆115Updated 4 months ago
- Summary of system papers/frameworks/codes/tools on training or serving large model☆56Updated last year
- 🔥 A minimal training framework for scaling FLA models☆107Updated 2 weeks ago