NonvolatileMemory / flash_tree_attn
☆16Updated last month
Alternatives and similar repositories for flash_tree_attn:
Users that are interested in flash_tree_attn are comparing it to the libraries listed below
- SQUEEZED ATTENTION: Accelerating Long Prompt LLM Inference☆36Updated 2 months ago
- The Official Implementation of Ada-KV: Optimizing KV Cache Eviction by Adaptive Budget Allocation for Efficient LLM Inference☆56Updated last week
- [ICLR 2024] SWIFT: On-the-Fly Self-Speculative Decoding for LLM Inference Acceleration☆33Updated last month
- The official implementation of paper: SimLayerKV: A Simple Framework for Layer-Level KV Cache Reduction.☆42Updated 3 months ago
- Official Implementation of SAM-Decoding: Speculative Decoding via Suffix Automaton☆16Updated last month
- [NeurIPS 2024] Fast Best-of-N Decoding via Speculative Rejection☆38Updated 3 months ago
- ☆36Updated 2 months ago
- This repo contains the source code for: Model Tells You What to Discard: Adaptive KV Cache Compression for LLMs☆32Updated 5 months ago
- Source code for the paper "LongGenBench: Long-context Generation Benchmark"☆14Updated 3 months ago
- ☆49Updated 8 months ago
- Pytorch implementation of our paper accepted by ICML 2024 -- CaM: Cache Merging for Memory-efficient LLMs Inference☆31Updated 7 months ago
- [ICLR 2025] TidalDecode: A Fast and Accurate LLM Decoding with Position Persistent Sparse Attention☆26Updated last month
- Multi-Candidate Speculative Decoding☆34Updated 9 months ago
- Fast and Robust Early-Exiting Framework for Autoregressive Language Models with Synchronized Parallel Decoding (EMNLP 2023 Long)☆56Updated 4 months ago
- Sirius, an efficient correction mechanism, which significantly boosts Contextual Sparsity models on reasoning tasks while maintaining its…☆21Updated 4 months ago
- QAQ: Quality Adaptive Quantization for LLM KV Cache☆45Updated 10 months ago
- Official Repo for SparseLLM: Global Pruning of LLMs (NeurIPS 2024)☆52Updated last month
- "Found in the Middle: How Language Models Use Long Contexts Better via Plug-and-Play Positional Encoding" Zhenyu Zhang, Runjin Chen, Shiw…☆25Updated 8 months ago
- OPT-Tree: Speculative Decoding with Adaptive Draft Tree Structure☆20Updated 5 months ago
- Quantized Attention on GPU☆34Updated 2 months ago
- ☆22Updated last year
- ☆49Updated last year
- Official PyTorch implementation of IntactKV: Improving Large Language Model Quantization by Keeping Pivot Tokens Intact☆38Updated 8 months ago
- 🔥 A minimal training framework for scaling FLA models☆32Updated this week
- Quantized Side Tuning: Fast and Memory-Efficient Tuning of Quantized Large Language Models☆41Updated 2 months ago
- The source code of "Merging Experts into One: Improving Computational Efficiency of Mixture of Experts (EMNLP 2023)":☆34Updated 9 months ago
- 16-fold memory access reduction with nearly no loss☆72Updated 2 months ago
- Implementation of "Decoding-time Realignment of Language Models", ICML 2024.☆17Updated 7 months ago
- Code associated with the paper **Draft & Verify: Lossless Large Language Model Acceleration via Self-Speculative Decoding**☆153Updated 8 months ago
- [ICLR 2025] PEARL: parallel speculative decoding with adaptive draft length☆32Updated 5 months ago