Predicting the success of Gradient Descent for a particular Dataset-Architecture-Initialization (DAI). - Umangi Jain; Harish G. Ramaswamy (CoRR, 2021)

Multimodal contrastive learning for remote sensing tasks. - Umangi Jain; Alex Wilson; Varun Gulshan (CoRR, 2022)

GaussianCut: Interactive segmentation via graph cut for 3D Gaussian Splatting. - Umangi Jain; Ashkan Mirzaei; Igor Gilitschenski (NeurIPS, 2024)

Satellite Sunroof: High-res Digital Surface Models and Roof Segmentation for Global Solar Mapping. - Vishal Batchu; Alex Wilson; Betty Peng; Carl Elkin; Umangi Jain; Christopher Van Arsdale; Ross Goroshin; Varun Gulshan (CoRR, 2024)