kyegomez / FlashAttention20Triton
Triton implementation of Flash Attention2.0
☆29Updated 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☆64Updated 8 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆64Updated 3 months ago
- Boosting 4-bit inference kernels with 2:4 Sparsity☆64Updated 5 months ago
- ☆100Updated 5 months ago
- Official repository for LightSeq: Sequence Level Parallelism for Distributed Training of Long Context Transformers☆205Updated 6 months ago
- ☆67Updated 2 months ago
- Ouroboros: Speculative Decoding with Large Model Enhanced Drafting (EMNLP 2024 main)☆84Updated 4 months ago
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆157Updated 7 months ago
- Vocabulary Parallelism☆17Updated 3 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆94Updated last month
- Summary of system papers/frameworks/codes/tools on training or serving large model☆56Updated last year
- [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding☆107Updated 2 months ago
- ☆81Updated 5 months ago
- ☆87Updated this week
- [NeurIPS 2024] The official implementation of "Kangaroo: Lossless Self-Speculative Decoding for Accelerating LLMs via Double Early Exitin…☆48Updated 7 months ago
- ☆111Updated this week
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆97Updated 7 months ago
- Decoding Attention is specially optimized for multi head attention (MHA) using CUDA core for the decoding stage of LLM inference.☆29Updated 3 months ago
- ☆62Updated 2 months ago
- ☆104Updated 6 months ago
- ☆76Updated 2 years ago
- 16-fold memory access reduction with nearly no loss☆77Updated this week
- [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference☆245Updated 2 months ago
- [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving☆295Updated 7 months ago
- ☆59Updated 2 weeks ago
- ☆61Updated 3 weeks ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆89Updated this week
- Quantized Attention on GPU☆34Updated 2 months ago
- ☆52Updated 10 months ago
- QQQ is an innovative and hardware-optimized W4A8 quantization solution for LLMs.☆101Updated this week