NonvolatileMemory / flash_tree_attn
☆19Updated 3 months ago
Alternatives and similar repositories for flash_tree_attn:
Users that are interested in flash_tree_attn are comparing it to the libraries listed below
- Quantized Attention on GPU☆45Updated 4 months ago
- SQUEEZED ATTENTION: Accelerating Long Prompt LLM Inference☆45Updated 4 months ago
- SKVQ: Sliding-window Key and Value Cache Quantization for Large Language Models☆17Updated 5 months ago
- More Tokens, Lower Precision: Towards the Optimal Token-Precision Trade-off in KV Cache Compression☆11Updated 2 months ago
- ☆19Updated 3 weeks ago
- [ICLR 2025] TidalDecode: A Fast and Accurate LLM Decoding with Position Persistent Sparse Attention☆31Updated last month
- Pytorch implementation of our paper accepted by ICML 2024 -- CaM: Cache Merging for Memory-efficient LLMs Inference☆35Updated 9 months ago
- XAttention: Block Sparse Attention with Antidiagonal Scoring☆118Updated this week
- Beyond KV Caching: Shared Attention for Efficient LLMs☆16Updated 8 months ago
- Source code for the paper "LongGenBench: Long-context Generation Benchmark"☆15Updated 5 months ago
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆66Updated this week
- LongSpec: Long-Context Speculative Decoding with Efficient Drafting and Verification☆42Updated last month
- This repo contains the source code for: Model Tells You What to Discard: Adaptive KV Cache Compression for LLMs☆35Updated 7 months ago
- The Official Implementation of Ada-KV: Optimizing KV Cache Eviction by Adaptive Budget Allocation for Efficient LLM Inference☆68Updated 2 months ago
- ☆72Updated last week
- Official PyTorch implementation of IntactKV: Improving Large Language Model Quantization by Keeping Pivot Tokens Intact☆43Updated 10 months ago
- [NeurIPS 2024] Fast Best-of-N Decoding via Speculative Rejection☆40Updated 5 months ago
- Multi-Candidate Speculative Decoding☆35Updated 11 months ago
- 16-fold memory access reduction with nearly no loss☆86Updated last week
- ☆47Updated 3 months ago
- ☆50Updated 10 months ago
- Activation-aware Singular Value Decomposition for Compressing Large Language Models☆59Updated 5 months ago
- Estimate MFU for DeepSeekV3☆21Updated 2 months ago
- A GPU-optimized system for efficient long-context LLMs decoding with low-bit KV cache.☆23Updated last week
- [ICLR 2025] PEARL: Parallel Speculative Decoding with Adaptive Draft Length☆67Updated 3 weeks ago
- [ICLR 2024] Jaiswal, A., Gan, Z., Du, X., Zhang, B., Wang, Z., & Yang, Y. Compressing llms: The truth is rarely pure and never simple.☆23Updated last year
- ☆20Updated this week
- [ICLR 2025] SWIFT: On-the-Fly Self-Speculative Decoding for LLM Inference Acceleration☆45Updated last month
- QAQ: Quality Adaptive Quantization for LLM KV Cache☆47Updated last year
- Sirius, an efficient correction mechanism, which significantly boosts Contextual Sparsity models on reasoning tasks while maintaining its…☆21Updated 6 months ago