Himanshu Jain

Researcher, Google

Joined

February 2021

Names

Himanshu Jain (Preferred)

Emails

****@google.com (Confirmed)

Personal Links

Google Scholar

ACL Anthology

Career & Education History

Researcher

Google (google.com)

2019 – Present

PhD student

Indian Institute of Technology Delhi, (iitd.ac.in)

2014 – 2019

Undergrad student

IIT Kanpur (iitk.ac.in)

2007 – 2012

Advisors, Relations & Conflicts

No relations added

Expertise

Deep Learning

Present

Large output spaces

Present

Publications

Efficient and Asymptotically Unbiased Constrained Decoding for Large Language Models
Haotian Ye, Himanshu Jain, Chong You, Ananda Theertha Suresh, Haowei Lin, James Zou, Felix Yu
- AISTATS 2025 Poster
- Readers: Everyone
Efficient Document Ranking with Learnable Late Interactions
Himanshu Jain, Ziwei Ji, Sashank J. Reddi, Ankit Singh Rawat, Felix Yu, Aditya Krishna Menon, Sadeep Jayasumana
- TF2M 2024 Poster
- Readers: Everyone
Efficient Document Ranking with Learnable Late Interactions
Himanshu Jain, Ziwei Ji, Ankit Singh Rawat, Andreas Veit, Sadeep Jayasumana, Sashank J. Reddi, Aditya Krishna Menon, Felix Yu
- WANT@ICML 2024 Poster
- Readers: Everyone
SpecTr: Fast Speculative Decoding via Optimal Transport
Ziteng Sun, Ananda Theertha Suresh, Jae Hun Ro, Ahmad Beirami, Himanshu Jain, Felix Yu, Michael Riley, Sanjiv Kumar
- ES-FoMO 2023 Poster
- Readers: Everyone
SpecTr: Fast Speculative Decoding via Optimal Transport
Ziteng Sun, Ananda Theertha Suresh, Jae Hun Ro, Ahmad Beirami, Himanshu Jain, Felix Yu
- NeurIPS 2023 poster
- Readers: Everyone
Teacher Guided Training: An Efficient Framework for Knowledge Transfer
Manzil Zaheer, Ankit Singh Rawat, Seungyeon Kim, Chong You, Himanshu Jain, Andreas Veit, Rob Fergus, Sanjiv Kumar
- Published: 01 Feb 2023, Last Modified: 14 Jan 2026
- ICLR 2023 poster
- Readers: Everyone
Treeformer: Dense Gradient Trees for Efficient Attention Computation
Lovish Madaan, Srinadh Bhojanapalli, Himanshu Jain, Prateek Jain
- Published: 01 Feb 2023, Last Modified: 01 Mar 2023
- ICLR 2023 poster
- Readers: Everyone
Do We Need Neural Collapse? Learning Diverse Features for Fine-grained and Long-tail Classification
Jiawei Ma, Chong You, Sashank J. Reddi, Sadeep Jayasumana, Himanshu Jain, Felix Yu, Shih-Fu Chang, Sanjiv Kumar
- Published: 01 Feb 2023, Last Modified: 13 Feb 2023
- Submitted to ICLR 2023
- Readers: Everyone
Leveraging Redundancy in Attention with Reuse Transformers
Srinadh Bhojanapalli, Ayan Chakrabarti, Andreas Veit, Michal Lukasik, Himanshu Jain, Frederick Liu, Yin-Wen Chang, Sanjiv Kumar
- 29 Sept 2021 (modified: 12 Oct 2025)
- ICLR 2022 Conference Withdrawn Submission
- Readers: Everyone

Co-Authors

View all 29 co-authors