activity
20202025
most citedSARATHI: Efficient LLM Inference by Piggybacking Decodes with Chunked Prefills

16 citations · 69 across the 20 of their papers we have counts for

collaborators

20 papers

cs.LG2025

Kascade: A Practical Sparse Attention Method for Long-Context LLM Inference

Dhruv Deshmukh, Saurabh Goyal, Nipun Kwatra +1

Attention is the dominant source of latency during long-context LLM inference, an increasingly popular workload with reasoning models and RAG. We propose Kascade, a training-free s…

cs.LG2025

On Evaluating Performance of LLM Inference Serving Systems

Amey Agrawal, Nitin Kedia, Anmol Agarwal +5

The rapid evolution of Large Language Model (LLM) inference systems has yielded significant efficiency improvements. However, our systematic analysis reveals that current evaluatio…

cs.DC2025

TokenWeave: Efficient Compute-Communication Overlap for Distributed LLM Inference

Raja Gond, Nipun Kwatra, Ramachandran Ramjee

Distributed inference of large language models (LLMs) using tensor parallelism can introduce communication overheads of % even over GPUs connected via NVLink, a high-speed GPU…

cs.LG2025

Niyama : Breaking the Silos of LLM Inference Serving

Kanishk Goel, Jayashree Mohan, Nipun Kwatra +2

The widespread adoption of Large Language Models (LLMs) has enabled diverse applications with very different latency requirements. Existing LLM serving frameworks rely on siloed in…

cs.LG2024★ 1 cited

Etalon: Holistic Performance Evaluation Framework for LLM Inference Systems

Amey Agrawal, Anmol Agarwal, Nitin Kedia +5

Serving large language models (LLMs) in production can incur substantial costs, which has prompted recent advances in inference system optimizations. Today, these systems are evalu…

cs.LG2024★ 1 cited

Accuracy is Not All You Need

Abhinav Dutta, Sanjeev Krishnan, Nipun Kwatra +1

When Large Language Models (LLMs) are compressed using techniques such as quantization, the predominant way to demonstrate the validity of such techniques is by measuring the model…

cs.LG2024★ 8 cited

Vidur: A Large-Scale Simulation Framework For LLM Inference

Amey Agrawal, Nitin Kedia, Jayashree Mohan +5

Optimizing the deployment of Large language models (LLMs) is expensive today since it requires experimentally running an application workload against an LLM implementation while ex…

cs.LG2024★ 15 cited

Taming Throughput-Latency Tradeoff in LLM Inference with Sarathi-Serve

Amey Agrawal, Nitin Kedia, Ashish Panwar +5

Each LLM serving request goes through two phases. The first is prefill which processes the entire input prompt and produces the first output token and the second is decode which ge…

cs.LG2023★ 16 cited

SARATHI: Efficient LLM Inference by Piggybacking Decodes with Chunked Prefills

Amey Agrawal, Ashish Panwar, Jayashree Mohan +3

Large Language Model (LLM) inference consists of two distinct phases - prefill phase which processes the input prompt and decode phase which generates output tokens autoregressivel…

eess.SP2023★ 1 cited

"Can't Take the Pressure?": Examining the Challenges of Blood Pressure Estimation via Pulse Wave Analysis

Suril Mehta, Nipun Kwatra, Mohit Jain +1

The use of observed wearable sensor data (e.g., photoplethysmograms [PPG]) to infer health measures (e.g., glucose level or blood pressure) is a very active area of research. Such…

cs.HC2022★ 2 cited

Towards Automating Retinoscopy for Refractive Error Diagnosis

Aditya Aggarwal, Siddhartha Gairola, Uddeshya Upadhyay +6

Refractive error is the most common eye disorder and is the key cause behind correctable visual impairment, responsible for nearly 80% of the visual impairment in the US. Refractiv…

cs.LG2022★ 1 cited

Distance Learner: Incorporating Manifold Prior to Model Training

Aditya Chetan, Nipun Kwatra

The manifold hypothesis (real world data concentrates near low-dimensional manifolds) is suggested as the principle behind the effectiveness of machine learning algorithms in very…

eess.IV2022★ 2 cited

Keratoconus Classifier for Smartphone-based Corneal Topographer

Siddhartha Gairola, Pallavi Joshi, Anand Balasubramaniam +3

Keratoconus is a severe eye disease that leads to deformation of the cornea. It impacts people aged 10-25 years and is the leading cause of blindness in that demography. Corneal to…

cs.DC2022★ 4 cited

Singularity: Planet-Scale, Preemptive and Elastic Scheduling of AI Workloads

Dharma Shukla, Muthian Sivathanu, Srinidhi Viswanatha +23

Lowering costs by driving high utilization across deep learning workloads is a crucial lever for cloud providers. We present Singularity, Microsoft's globally distributed schedulin…

cs.DC2021★ 2 cited

Varuna: Scalable, Low-cost Training of Massive Deep Learning Models

Sanjith Athlur, Nitika Saran, Muthian Sivathanu +2

Systems for training massive deep learning models (billions of parameters) today assume and require specialized "hyper-clusters": hundreds or thousands of GPUs wired with specializ…

cs.HC2021★ 1 cited

SmartKC: Smartphone-based Corneal Topographer for Keratoconus Detection

Siddhartha Gairola, Murtuza Bohra, Nadeem Shaheer +6

Keratoconus is a severe eye disease affecting the cornea (the clear, dome-shaped outer surface of the eye), causing it to become thin and develop a conical bulge. The diagnosis of…

cs.LG2021

LRTuner: A Learning Rate Tuner for Deep Neural Networks

Nikhil Iyer, V Thejas, Nipun Kwatra +2

One very important hyperparameter for training deep neural networks is the learning rate schedule of the optimizer. The choice of learning rate schedule determines the computationa…

cs.SD2020

RespireNet: A Deep Neural Network for Accurately Detecting Abnormal Lung Sounds in Limited Data Setting

Siddhartha Gairola, Francis Tom, Nipun Kwatra +1

Auscultation of respiratory sounds is the primary tool for screening and diagnosing lung diseases. Automated analysis, coupled with digital stethoscopes, can play a crucial role in…

cs.LG2020★ 1 cited

S2cGAN: Semi-Supervised Training of Conditional GANs with Fewer Labels

Arunava Chakraborty, Rahul Ragesh, Mahir Shah +1

Generative adversarial networks (GANs) have been remarkably successful in learning complex high dimensional real word distributions and generating realistic samples. However, they…

cs.LG2020★ 14 cited

Wide-minima Density Hypothesis and the Explore-Exploit Learning Rate Schedule

Nikhil Iyer, V Thejas, Nipun Kwatra +2

Several papers argue that wide minima generalize better than narrow minima. In this paper, through detailed experiments that not only corroborate the generalization properties of w…