← Search

Rohan Shah

2 accepted papers

2024

Scaling transformer neural networks for skillful and reliable medium-range weather forecasting

NeurIPS 2024poster

Weather forecasting is a fundamental problem for anticipating and mitigating the impacts of climate change. Recently, data-driven approaches for weather forecasting based on deep learning have shown great promise, achieving accuracies that are competitive with operational systems. However, those met…

2022

PAC Mode Estimation using PPR Martingale Confidence Sequences

AISTATS 2022poster

We consider the problem of correctly identifying the mode of a discrete distribution $\mathcal{P}$ with sufficiently high probability by observing a sequence of i.i.d. samples drawn from $\mathcal{P}$. This problem reduces to the estimation of a single parameter when $\mathcal{P}$ has a support set…