Rajit Rajpal

I am currently a PhD student at the University of Edinburgh in the MAC-MIGS programme funded by the Probabilistic AI Hub. My interests lie in generative modeling and optimization.

Supervised by Benedict Leimkuhler and Nikolay Malkin.

Email / Github / LinkedIn / Google Scholar

Projects
	Depth Guided Diffusion World Models Coming soon (2026)
	Adaptive Momentum and Nonlinear Damping in Neural Network Training Katerina Karoni, Rajit Rajpal, Benedict Leimkuhler, Gabriel Stoltz Arxiv Preprint (In Review) pdf / code
	Adaptive Stepsizing for Stochastic Gradient Langevin Dynamics in Bayesian Neural Networks Rajit Rajpal, Benedict Leimkuhler, Yuanhao Jiang AAAI 2025 Workshop @ Math4AI (Oral, Top 10%) pdf / code
	Frugal Zero-Shot Diffusion-Based Image Restoration Rajit Rajpal, Marcos Obando, Dolly Chen, Bernardin Amogou Runner-up @ Diffusion Models PhD Hackathon (2024) poster / code
	Does Translation Invariance Improve Tractable Generative Models? Rajit Rajpal, Antonio Vergari MSc Dissertation; Poster @ BayesAI Workshop (2024) pdf / code

Teaching

I have been a teaching assistant for:

Education

	University of Edinburgh Aug 2023 – Aug 2028 MSc/PhD in Applied Mathematics
	University of California, Berkeley Aug 2019 – May 2022 B.A. Data Science

Experience

	Overworld AI Feb 2026 - Present Member of Technical Staff, Pretraining
	Qubit Analytics May – Dec 2025 Research Scientist, Video Generation
	NASA Ames Research Center – Rhombus Power Mar 2022 – Mar 2023 Data Scientist

Cloned from here!
Last updated: Nov 18, 2025