| Checkmate: Breaking the memory wall with optimal tensor rematerialization P Jain, A Jain, A Nrusimha, A Gholami, P Abbeel, J Gonzalez, K Keutzer, ... Proceedings of Machine Learning and Systems 2, 497-511, 2020 | 206 | 2020 |
| Self-supervised pretraining improves self-supervised pretraining CJ Reed, X Yue, A Nrusimha, S Ebrahimi, V Vijaykumar, R Mao, B Li, ... Proceedings of the IEEE/CVF Winter Conference on Applications of Computer …, 2022 | 126 | 2022 |
| Reducing transformer key-value cache size with cross-layer attention W Brandon, M Mishra, A Nrusimha, R Panda, J Ragan-Kelley Advances in Neural Information Processing Systems 37, 86927-86957, 2025 | 35 | 2025 |
| Integer-only zero-shot quantization for efficient speech recognition S Kim, A Gholami, Z Yao, N Lee, P Wang, A Nrusimha, B Zhai, T Gao, ... ICASSP 2022-2022 IEEE International Conference on Acoustics, Speech and …, 2022 | 27 | 2022 |
| Hydra: Sequentially-dependent draft heads for medusa decoding Z Ankner, R Parthasarathy, A Nrusimha, C Rinard, J Ragan-Kelley, ... arXiv preprint arXiv:2402.05109, 2024 | 24 | 2024 |
| Striped attention: Faster ring attention for causal transformers W Brandon, A Nrusimha, K Qian, Z Ankner, T Jin, Z Song, J Ragan-Kelley arXiv preprint arXiv:2311.09431, 2023 | 24 | 2023 |
| Sagedb: An instance-optimized data analytics system J Ding, R Marcus, A Kipf, V Nathan, A Nrusimha, K Vaidya, A van Renen, ... Proceedings of the VLDB Endowment 15 (13), 2022 | 16 | 2022 |
| Towards verifiable text generation with symbolic references LT Hennigen, S Shen, A Nrusimha, B Gapp, D Sontag, Y Kim arXiv preprint arXiv:2311.09188, 2023 | 8 | 2023 |
| Mitigating the impact of outlier channels for language model quantization with activation regularization A Nrusimha, M Mishra, N Wang, D Alistarh, R Panda, Y Kim arXiv preprint arXiv:2404.03605, 2024 | 5 | 2024 |
| Predicting memory-based consumer choices from recall and preferences Z Zhang, A Nrusimha, M Hsu Advances in Consumer Research 46, 878-879, 2018 | 2 | 2018 |
| Predicting consumer brand recall and choice using largescale text corpora Z Zhang, A Nrusimha, M Hsu Advances in Consumer Research 46, 204-208, 2018 | 2 | 2018 |