Gaurav Jain

Researcher, d-Matrix

Joined

October 2023

Names

Gaurav Jain (Preferred)

Emails

****@d-matrix.ai (Confirmed)

,

****@gmail.com (Confirmed)

Personal Links

Career & Education History

Researcher

d-Matrix (d-matrix.ai)

2023 – Present

MS student

University of Wisconsin - Madison (wisc.edu)

2018 – 2020

Advisors, Relations & Conflicts

Coworker

2024 – Present

Coworker

Akhil Arunkumar

2023 – Present

Coauthor

Prashant J. Nair

2023 – Present

Coworker

2023 – Present

Expertise

No areas of expertise listed

Publications

Keyformer: KV Cache reduction through key tokens selection for Efficient Generative Inference
Muhammad Adnan, Akhil Arunkumar, Gaurav Jain, Prashant J. Nair, Ilya Soloveychik, Purushotham Kamath
- Proceedings of Machine Learning and Systems
- Readers: Everyone
Accelerating LLM Inference with Lossless Speculative Decoding Algorithms for Heterogeneous Vocabularies
Nadav Timor, Jonathan Mamou, Daniel Korat, Moshe Berchansky, Gaurav Jain, Oren Pereg, Moshe Wasserblat, David Harel
- ICML 2025 oral
- Readers: Everyone
Dialogue Without Limits: Constant-Sized KV Caches for Extended Response in LLMs
Ravi Ghadia, Avinash Kumar, Gaurav Jain, Prashant J. Nair, Poulami Das
- ICML 2025 poster
- Readers: Everyone

Co-Authors