Shah, Rohan

5 publications

NeurIPS 2024 Scaling Transformer Neural Networks for Skillful and Reliable Medium-Range Weather Forecasting Tung Nguyen, Rohan Shah, Hritik Bansal, Troy Arcomano, Romit Maulik, Veerabhadra Kotamarthi, Ian Foster, Sandeep Madireddy, Aditya Grover

ICLRW 2024 Scaling Transformers for Skillful and Reliable Medium-Range Weather Forecasting Tung Nguyen, Rohan Shah, Hritik Bansal, Troy Arcomano, Sandeep Madireddy, Romit Maulik, Veerabhadra Kotamarthi, Ian Foster, Aditya Grover

NeurIPSW 2023 Correlated Trajectory Uncertainty for Adaptive Sequential Decision Making Ian Char, Youngseog Chung, Rohan Shah, Willie Neiswanger, Jeff Schneider

NeurIPSW 2023 Turn Down the Noise: Leveraging Diffusion Models for Test-Time Adaptation via Pseudo-Label Ensembling Mrigank Raman, Rohan Shah, Akash Kannan, Pranit Chawla

AISTATS 2022 PAC Mode Estimation Using PPR Martingale Confidence Sequences Shubham Anand Jain, Rohan Shah, Sanit Gupta, Denil Mehta, Inderjeet J. Nair, Jian Vora, Sushil Khyalia, Sourav Das, Vinay J. Ribeiro, Shivaram Kalyanakrishnan