Low-Memory Neural Network Training: A Technical Report. - Nimit Sharad Sohoni; Christopher Richard Aberger; Megan Leszczynski; Jian Zhang 049; Christopher Ré (CoRR, 2019)

Ivy: Instrumental Variable Synthesis for Causal Inference. - Zhaobin Kuang; Frederic Sala; Nimit Sharad Sohoni; Sen Wu 002; Aldo Córdova-Palomera; Jared Dunnmon; James Priest; Christopher Ré (AISTATS, 2020)

Near-Optimal Methods for Minimizing Star-Convex Functions and Beyond. - Oliver Hinder; Aaron Sidford; Nimit Sharad Sohoni (COLT, 2020)

Kaleidoscope: An Efficient, Learnable Representation For All Structured Linear Maps. - Tri Dao; Nimit Sharad Sohoni; Albert Gu; Matthew Eichhorn; Amit Blonder; Megan Leszczynski; Atri Rudra; Christopher Ré (ICLR, 2020)

No Subclass Left Behind: Fine-Grained Robustness in Coarse-Grained Classification Problems. - Nimit Sharad Sohoni; Jared Dunnmon; Geoffrey Angus; Albert Gu; Christopher Ré (NeurIPS, 2020)

Low-Shot Validation: Active Importance Sampling for Estimating Classifier Performance on Rare Categories. - Fait Poms; Vishnu Sarukkai; Ravi Teja Mullapudi; Nimit Sharad Sohoni; William R. Mark; Deva Ramanan; Kayvon Fatahalian (ICCV, 2021)

Mandoline: Model Evaluation under Distribution Shift. - Mayee F. Chen; Karan Goel; Nimit Sharad Sohoni; Fait Poms; Kayvon Fatahalian; Christopher Ré (ICML, 2021)

Observational Supervision for Medical Image Classification Using Gaze Data. - Khaled Saab 002; Sarah M. Hooper; Nimit Sharad Sohoni; Jupinder Parmar; Brian Pogatchnik; Sen Wu 002; Jared A. Dunnmon; Hongyang R. Zhang; Daniel L. Rubin; Christopher Ré (MICCAI (2), 2021)

Monarch: Expressive Structured Matrices for Efficient and Accurate Training. - Tri Dao; Beidi Chen; Nimit Sharad Sohoni; Arjun D. Desai; Michael Poli; Jessica Grogan; Alexander Liu; Aniruddh Rao; Atri Rudra; Christopher Ré (ICML, 2022)

Correct-N-Contrast: a Contrastive Approach for Improving Robustness to Spurious Correlations. - Michael Zhang; Nimit Sharad Sohoni; Hongyang R. Zhang; Chelsea Finn; Christopher Ré (ICML, 2022)

BARACK: Partially Supervised Group Robustness With Guarantees. - Nimit Sharad Sohoni; Maziar Sanjabi; Nicolas Ballas; Aditya Grover; Shaoliang Nie; Hamed Firooz; Christopher Ré (CoRR, 2022)

Llamba: Scaling Distilled Recurrent Models for Efficient Language Processing. - Aviv Bick; Tobias Katsch; Nimit Sharad Sohoni; Arjun Desai; Albert Gu (CoRR, 2025)