kyegomez / FlashAttention20Triton
Triton implementation of Flash Attention2.0
☆31Updated last year
Alternatives and similar repositories for FlashAttention20Triton:
Users that are interested in FlashAttention20Triton are comparing it to the libraries listed below
- Odysseus: Playground of LLM Sequence Parallelism☆68Updated 9 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆77Updated 4 months ago
- ☆67Updated 2 months ago
- ☆75Updated this week
- An easy-to-use package for implementing SmoothQuant for LLMs☆95Updated 10 months ago
- Boosting 4-bit inference kernels with 2:4 Sparsity☆71Updated 6 months ago
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆157Updated 8 months ago
- [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding☆111Updated 3 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆111Updated 2 months ago
- QQQ is an innovative and hardware-optimized W4A8 quantization solution for LLMs.☆109Updated 2 weeks ago
- ☆32Updated 7 months ago
- ☆145Updated last year
- Ouroboros: Speculative Decoding with Large Model Enhanced Drafting (EMNLP 2024 main)☆90Updated last week
- ☆102Updated 7 months ago
- Summary of system papers/frameworks/codes/tools on training or serving large model☆56Updated last year
- [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving☆299Updated 8 months ago
- Official PyTorch implementation of FlatQuant: Flatness Matters for LLM Quantization☆112Updated this week
- ☆118Updated last month
- [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training☆168Updated last month
- [ICLR 2025] PEARL: Parallel Speculative Decoding with Adaptive Draft Length☆62Updated 2 weeks ago
- An algorithm for weight-activation quantization (W4A4, W4A8) of LLMs, supporting both static and dynamic quantization☆123Updated last month
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆66Updated this week
- 🔥 A minimal training framework for scaling FLA models☆82Updated last week
- QAQ: Quality Adaptive Quantization for LLM KV Cache☆47Updated last year
- [ACL 2024] A novel QAT with Self-Distillation framework to enhance ultra low-bit LLMs.☆109Updated 10 months ago
- 16-fold memory access reduction with nearly no loss☆83Updated this week
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆102Updated 8 months ago
- ☆233Updated 10 months ago
- Fast Hadamard transform in CUDA, with a PyTorch interface☆154Updated 10 months ago
- Decoding Attention is specially optimized for MHA, MQA, GQA and MLA using CUDA core for the decoding stage of LLM inference.☆35Updated 3 weeks ago