Mahankali, Arvind V.

3 publications

NeurIPSW 2024 Formal Theorem Proving by Rewarding LLMs to Decompose Proofs Hierarchically Kefan Dong, Arvind V. Mahankali, Tengyu Ma

ICLR 2024 One Step of Gradient Descent Is Provably the Optimal In-Context Learner with One Layer of Linear Self-Attention Arvind V. Mahankali, Tatsunori Hashimoto, Tengyu Ma

ICML 2021 Streaming and Distributed Algorithms for Robust Column Subset Selection Shuli Jiang, Dennis Li, Irene Mengze Li, Arvind V Mahankali, David Woodruff