Tandem Transformers for Inference Efficient LLMs. - Aishwarya P. S.; Pranav Ajit Nair; Yashas Samaga; Toby Boyd; Sanjiv Kumar; Prateek Jain 002; Praneeth Netrapalli (ICML, 2024)

All Mistakes are not Equal: Comprehensive Hierarchy Aware Multilabel Predictions (CHAMP). - Ashwin Vaswani; Yashas Samaga; Gaurav Aggarwal; Praneeth Netrapalli; Narayan G. Hegde (ICPR (1), 2024)

HiRE: High Recall Approximate Top-k Estimation for Efficient LLM Inference. - Yashas Samaga; Varun Yerram; Chong You; Srinadh Bhojanapalli; Sanjiv Kumar; Prateek Jain 002; Praneeth Netrapalli (CoRR, 2024)

Faster Approx. Top-K: Harnessing the Full Power of Two Stages. - Yashas Samaga; Varun Yerram; Spandana Raj Babbula; Prateek Jain 002; Praneeth Netrapalli (CoRR, 2025)