Checkmate: Breaking the Memory Wall with Optimal Tensor Rematerialization. - Paras Jain 001; Ajay Jain; Aniruddha Nrusimha; Amir Gholami; Pieter Abbeel; Kurt Keutzer; Ion Stoica; Joseph Gonzalez 001 (MLSys, 2020)

Q-ASR: Integer-only Zero-shot Quantization for Efficient Speech Recognition. - Sehoon Kim 001; Amir Gholami; Zhewei Yao; Aniruddha Nrusimha; Bohan Zhai; Tianren Gao; Michael W. Mahoney; Kurt Keutzer (CoRR, 2021)

Integer-Only Zero-Shot Quantization for Efficient Speech Recognition. - Sehoon Kim 001; Amir Gholami; Zhewei Yao; Nicholas Lee; Patrick Wang; Aniruddha Nrusimha; Bohan Zhai; Tianren Gao; Michael W. Mahoney; Kurt Keutzer (ICASSP, 2022)

Self-Supervised Pretraining Improves Self-Supervised Pretraining. - Colorado J. Reed; Xiangyu Yue 001; Ani Nrusimha; Sayna Ebrahimi; Vivek Vijaykumar; Richard Mao; Bo Li 080; Shanghang Zhang; Devin Guillory; Sean Metzger; Kurt Keutzer; Trevor Darrell (WACV, 2022)

SageDB: An Instance-Optimized Data Analytics System. - Jialin Ding 001; Ryan Marcus; Andreas Kipf; Vikram Nathan; Aniruddha Nrusimha; Kapil Vaidya; Alexander van Renen; Tim Kraska (Proc. VLDB Endow., 2022)

Towards Verifiable Text Generation with Symbolic References. - Lucas Torroba Hennigen; Shannon Shen 001; Aniruddha Nrusimha; Bernhard Gapp; David A. Sontag; Yoon Kim (CoRR, 2023)

Striped Attention: Faster Ring Attention for Causal Transformers. - William Brandon; Aniruddha Nrusimha; Kevin Qian; Zachary Ankner; Tian Jin; Zhiye Song; Jonathan Ragan-Kelley (CoRR, 2023)

Reducing Transformer Key-Value Cache Size with Cross-Layer Attention. - William Brandon; Mayank Mishra; Aniruddha Nrusimha; Rameswar Panda; Jonathan Ragan-Kelley (NeurIPS, 2024)

Hydra: Sequentially-Dependent Draft Heads for Medusa Decoding. - Zachary Ankner; Rishab Parthasarathy; Aniruddha Nrusimha; Christopher Rinard; Jonathan Ragan-Kelley; William Brandon (CoRR, 2024)

Mitigating the Impact of Outlier Channels for Language Model Quantization with Activation Regularization. - Aniruddha Nrusimha; Mayank Mishra; Naigang Wang; Dan Alistarh; Rameswar Panda; Yoon Kim (CoRR, 2024)

FlashFormer: Whole-Model Kernels for Efficient Low-Batch Inference. - Aniruddha Nrusimha; William Brandon; Mayank Mishra; Yikang Shen; Rameswar Panda; Jonathan Ragan-Kelley; Yoon Kim (CoRR, 2025)