LeanDojo: Theorem Proving with Retrieval-Augmented Language Models. - Kaiyu Yang; Aidan M. Swope; Alex Gu; Rahul Chalamala; Peiyang Song 002; Shixing Yu; Saad Godil; Ryan J. Prenger; Animashree Anandkumar (NeurIPS, 2023)

RedPajama: an Open Dataset for Training Large Language Models. - Maurice Weber; Daniel Y. Fu; Quentin Anthony; Yonatan Oren; Shane Adams; Anton Alexandrov; Xiaozhong Lyu; Huu Nguyen; Xiaozhe Yao; Virginia Adams; Ben Athiwaratkun; Rahul Chalamala; et al. (NeurIPS, 2024)

Dragonfly: Multi-Resolution Zoom Supercharges Large Visual-Language Model. - Kezhen Chen; Rahul Thapa; Rahul Chalamala; Ben Athiwaratkun; Shuaiwen Leon Song; James Zou 001 (CoRR, 2024)

LoLCATs: On Low-Rank Linearizing of Large Language Models. - Michael Zhang; Simran Arora; Rahul Chalamala; Benjamin Frederick Spector; Alan Wu; Krithik Ramesh; Aaryan Singhal; Christopher Ré (ICLR, 2025)

SMIR: Efficient Synthetic Data Pipeline To Improve Multi-Image Reasoning. - Andrew Li; Rahul Thapa; Rahul Chalamala; Qingyang Wu; Kezhen Chen; James Zou (CoRR, 2025)