Sanjeev Raja

University of California, Berkeley

I’m a second-year Computer Science PhD student at Berkeley AI Research (BAIR) advised by Aditi Krishnapriyan.

My interests lie at the intersection of machine learning and the molecular sciences, with a particular focus on developing ML methods to accelerate molecular dynamics simulations. More broadly, I am interested in combining the expressivity of modern deep neural networks with the generalization capabilities of first-principles scientific models to improve scientific computing workflows.

In the past, I worked on ML for climate simulations. I have been a research intern at Lawrence Berkeley National Laboratory, where I worked with Jaideep Pathak and Anima Anandkumar on FourCastNet, which was a state-of-the-art weather prediction model at the time. I also represented my university in ProjectX, an international research competition on ML for climate change solutions, for which we developed adversarial super-resolution methods for global climate simulations.

news

Feb 21, 2024 Check out our preprint and talk introducing StABlE, a multi-modal approach for training stable machine learning force fields that combines supervision from quantum-mechanical energies/forces and system observables!
Aug 24, 2022 Starting my PhD in Computer Science at UC Berkeley. Excited for the next chapter!
Apr 1, 2022 Check out our preprint on using Adaptive Fourier Neural Operators for global weather forecasting.
Jan 10, 2021 Won Best Paper Award and $20,000 prize at the ProjectX research competition and presented at the UofT AI Conference. Check out our paper on super-resolution of global climate models.