thunlp / OuroborosLinks
Ouroboros: Speculative Decoding with Large Model Enhanced Drafting (EMNLP 2024 main)
☆109Updated 5 months ago
Alternatives and similar repositories for Ouroboros
Users that are interested in Ouroboros are comparing it to the libraries listed below
Sorting:
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆165Updated last year
- [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding☆124Updated 9 months ago
- Code associated with the paper **Draft & Verify: Lossless Large Language Model Acceleration via Self-Speculative Decoding**☆201Updated 7 months ago
- ☆278Updated 2 months ago
- Multi-Candidate Speculative Decoding☆36Updated last year
- [NeurIPS 2024] The official implementation of "Kangaroo: Lossless Self-Speculative Decoding for Accelerating LLMs via Double Early Exitin…☆59Updated last year
- REST: Retrieval-Based Speculative Decoding, NAACL 2024☆207Updated this week
- 16-fold memory access reduction with nearly no loss☆105Updated 5 months ago
- Repository of LV-Eval Benchmark☆70Updated last year
- Official Implementation of SAM-Decoding: Speculative Decoding via Suffix Automaton☆32Updated 7 months ago
- ☆141Updated 7 months ago
- Implementation for FP8/INT8 Rollout for RL training without performence drop.☆187Updated 2 weeks ago
- The official implementation of paper: SimLayerKV: A Simple Framework for Layer-Level KV Cache Reduction.☆48Updated 10 months ago
- [NeurIPS 2024] Fast Best-of-N Decoding via Speculative Rejection☆51Updated 10 months ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆216Updated last year
- An innovative method expediting LLMs via streamlined semi-autoregressive generation and draft verification.☆26Updated 5 months ago
- Layer-Condensed KV cache w/ 10 times larger batch size, fewer params and less computation. Dramatic speed up with better task performance…☆155Updated 5 months ago
- ☆118Updated 3 months ago
- [ICLR 2025] PEARL: Parallel Speculative Decoding with Adaptive Draft Length☆111Updated 5 months ago
- ☆45Updated 9 months ago
- Boosting 4-bit inference kernels with 2:4 Sparsity☆82Updated last year
- Implementation of NAACL 2024 Outstanding Paper "LM-Infinite: Simple On-the-Fly Length Generalization for Large Language Models"☆149Updated 6 months ago
- ☆43Updated 3 months ago
- SeerAttention: Learning Intrinsic Sparse Attention in Your LLMs☆151Updated last month
- The Official Implementation of Ada-KV: Optimizing KV Cache Eviction by Adaptive Budget Allocation for Efficient LLM Inference☆92Updated 2 months ago
- Simple extension on vLLM to help you speed up reasoning model without training.☆188Updated 3 months ago
- Easy control for Key-Value Constrained Generative LLM Inference(https://arxiv.org/abs/2402.06262)☆63Updated last year
- [ICLR 2024 Spotlight] Code for the paper "Merge, Then Compress: Demystify Efficient SMoE with Hints from Its Routing Policy"☆93Updated 2 months ago
- Async pipelined version of Verl☆116Updated 5 months ago
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆140Updated 3 months ago