NoakLiu / PiKVLinks
PiKV: KV Cache Management System for Mixture of Experts [Efficient ML System]
☆42Updated 3 weeks ago
Alternatives and similar repositories for PiKV
Users that are interested in PiKV are comparing it to the libraries listed below
Sorting:
- ☆18Updated 8 months ago
- ☆50Updated 5 months ago
- Quantized Attention on GPU☆44Updated 11 months ago
- [ACL 2025] Squeezed Attention: Accelerating Long Prompt LLM Inference☆54Updated 11 months ago
- Beyond KV Caching: Shared Attention for Efficient LLMs☆19Updated last year
- [ICLR 2025] TidalDecode: A Fast and Accurate LLM Decoding with Position Persistent Sparse Attention☆48Updated 3 months ago
- [DAC'25] Official implement of "HybriMoE: Hybrid CPU-GPU Scheduling and Cache Management for Efficient MoE Inference"☆85Updated 4 months ago
- FastCache: Fast Caching for Diffusion Transformer Through Learnable Linear Approximation [Efficient ML Model]☆44Updated 2 months ago
- [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding☆130Updated 11 months ago
- DeepGEMM: clean and efficient FP8 GEMM kernels with fine-grained scaling☆21Updated 2 weeks ago
- [ICML24] Pruner-Zero: Evolving Symbolic Pruning Metric from scratch for LLMs☆94Updated 11 months ago
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆169Updated last year
- [CoLM'25] The official implementation of the paper <MoA: Mixture of Sparse Attention for Automatic Large Language Model Compression>☆148Updated 3 months ago
- Fast and memory-efficient exact kmeans☆116Updated last week
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆124Updated 4 months ago
- ☆130Updated 5 months ago
- [ACL 2024] RelayAttention for Efficient Large Language Model Serving with Long System Prompts☆40Updated last year
- 16-fold memory access reduction with nearly no loss☆106Updated 7 months ago
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆151Updated 3 weeks ago
- Compression for Foundation Models☆35Updated 3 months ago
- PipeRAG: Fast Retrieval-Augmented Generation via Algorithm-System Co-design (KDD 2025)☆26Updated last year
- Pytorch implementation of our paper accepted by ICML 2024 -- CaM: Cache Merging for Memory-efficient LLMs Inference☆47Updated last year
- 🔥 LLM-powered GPU kernel synthesis: Train models to convert PyTorch ops into optimized Triton kernels via SFT+RL. Multi-turn compilation…☆94Updated last month
- ☆83Updated 9 months ago
- [ICML 2025] SliM-LLM: Salience-Driven Mixed-Precision Quantization for Large Language Models☆45Updated last year
- Accelerate LLM preference tuning via prefix sharing with a single line of code☆51Updated 4 months ago
- Code repo for efficient quantized MoE inference with mixture of low-rank compensators☆27Updated 6 months ago
- ☆103Updated 5 months ago
- ☆15Updated 7 months ago
- ☆120Updated 2 months ago