← Search

Gaurav Jain

2 accepted papers

2025

Accelerating LLM Inference with Lossless Speculative Decoding Algorithms for Heterogeneous Vocabularies

ICML 2025oral

Accelerating the inference of large language models (LLMs) is a critical challenge in generative AI. Speculative decoding (SD) methods offer substantial efficiency gains by generating multiple tokens using a single target forward pass. However, existing SD approaches require the drafter and target m…

Cited by 0SourcePDFScholar
2025

Dialogue Without Limits: Constant-Sized KV Caches for Extended Response in LLMs

ICML 2025poster

Autoregressive Transformers rely on Key-Value (KV) caching to accelerate inference. However, the linear growth of the KV cache with context length leads to excessive memory consumption and bandwidth constraints. Existing methods drop distant tokens or compress states in a lossy manner, sacrificing a…

Cited by 1SourcePDFScholar