NoakLiu / PiKVLinks
PiKV: KV Cache Management System for Mixture of Experts [Efficient ML System]
☆48Updated 3 months ago
Alternatives and similar repositories for PiKV
Users that are interested in PiKV are comparing it to the libraries listed below
Sorting:
- [DAC'25] Official implement of "HybriMoE: Hybrid CPU-GPU Scheduling and Cache Management for Efficient MoE Inference"☆99Updated last month
- [ICLR 2025] TidalDecode: A Fast and Accurate LLM Decoding with Position Persistent Sparse Attention☆52Updated 5 months ago
- ☆52Updated 8 months ago
- Quantized Attention on GPU☆44Updated last year
- ☆22Updated 10 months ago
- PipeRAG: Fast Retrieval-Augmented Generation via Algorithm-System Co-design (KDD 2025)☆30Updated last year
- [ACL 2025] Squeezed Attention: Accelerating Long Prompt LLM Inference☆56Updated last year
- 16-fold memory access reduction with nearly no loss☆109Updated 10 months ago
- Accelerating Large-Scale Reasoning Model Inference with Sparse Self-Speculative Decoding☆82Updated 2 months ago
- Beyond KV Caching: Shared Attention for Efficient LLMs☆20Updated last year
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆160Updated 3 months ago
- Artifact for "Marconi: Prefix Caching for the Era of Hybrid LLMs" [MLSys '25 Outstanding Paper Award, Honorable Mention]☆49Updated 10 months ago
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆176Updated last year
- Efficient Long-context Language Model Training by Core Attention Disaggregation☆85Updated this week
- [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding☆137Updated last year
- Pytorch implementation of our paper accepted by ICML 2024 -- CaM: Cache Merging for Memory-efficient LLMs Inference☆48Updated last year
- AdaSkip: Adaptive Sublayer Skipping for Accelerating Long-Context LLM Inference☆20Updated last year
- ☆117Updated 8 months ago
- [NeurIPS'25 Spotlight] Adaptive Attention Sparsity with Hierarchical Top-p Pruning☆87Updated 2 months ago
- [ICLR 2025] DeFT: Decoding with Flash Tree-attention for Efficient Tree-structured LLM Inference☆48Updated 7 months ago
- An auxiliary project analysis of the characteristics of KV in DiT Attention.☆32Updated last year
- ☆36Updated 11 months ago
- DeeperGEMM: crazy optimized version☆73Updated 8 months ago
- Scalable long-context LLM decoding that leverages sparsity—by treating the KV cache as a vector storage system.☆119Updated last month
- [HPCA 2026] A GPU-optimized system for efficient long-context LLMs decoding with low-bit KV cache.☆79Updated last month
- [ICLR'25] Fast Inference of MoE Models with CPU-GPU Orchestration☆258Updated last year
- DeepGEMM: clean and efficient FP8 GEMM kernels with fine-grained scaling☆21Updated last month
- QJL: 1-Bit Quantized JL transform for KV Cache Quantization with Zero Overhead☆31Updated last year
- 🔥 LLM-powered GPU kernel synthesis: Train models to convert PyTorch ops into optimized Triton kernels via SFT+RL. Multi-turn compilation…☆116Updated 2 months ago
- Compression for Foundation Models☆34Updated 6 months ago