li-plus / flash-preferenceLinks
Accelerate LLM preference tuning via prefix sharing with a single line of code
☆41Updated last month
Alternatives and similar repositories for flash-preference
Users that are interested in flash-preference are comparing it to the libraries listed below
Sorting:
- Quantized Attention on GPU☆44Updated 7 months ago
- ☆49Updated last month
- Odysseus: Playground of LLM Sequence Parallelism☆70Updated last year
- ☆71Updated last month
- ☆60Updated 2 months ago
- ☆39Updated this week
- DeeperGEMM: crazy optimized version☆69Updated last month
- ☆114Updated 3 weeks ago
- Estimate MFU for DeepSeekV3☆24Updated 5 months ago
- [ACL 2024] RelayAttention for Efficient Large Language Model Serving with Long System Prompts☆40Updated last year
- ☆21Updated 3 months ago
- A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters☆47Updated 11 months ago
- ☆77Updated 2 months ago
- 16-fold memory access reduction with nearly no loss☆99Updated 3 months ago
- Transformers components but in Triton☆34Updated last month
- [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding☆116Updated 6 months ago
- Beyond KV Caching: Shared Attention for Efficient LLMs☆19Updated 11 months ago
- patches for huggingface transformers to save memory☆23Updated 3 weeks ago
- The official implementation of paper: SimLayerKV: A Simple Framework for Layer-Level KV Cache Reduction.☆46Updated 8 months ago
- An auxiliary project analysis of the characteristics of KV in DiT Attention.☆31Updated 6 months ago
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆113Updated last month
- Best practices for testing advanced Mixtral, DeepSeek, and Qwen series MoE models using Megatron Core MoE.☆21Updated 2 weeks ago
- ☆130Updated 4 months ago
- Decoding Attention is specially optimized for MHA, MQA, GQA and MLA using CUDA core for the decoding stage of LLM inference.☆38Updated 2 weeks ago
- ☆33Updated this week
- [ICLR 2025] DeFT: Decoding with Flash Tree-attention for Efficient Tree-structured LLM Inference☆24Updated last week
- ☆63Updated this week
- Flash-Muon: An Efficient Implementation of Muon Optimizer☆131Updated last week
- ☆86Updated 3 months ago
- ☆19Updated 6 months ago