Rajit Rajpal

I am currently a PhD student at the University of Edinburgh in the MAC-MIGS programme funded by the ProbAI Hub and supervised by Benedict Leimkuhler and Esmeralda S. Whitammer. I am also a Founding Research Scientist at Overworld building Diffusion World Models.

Research
	Waypoint-2: A Separable Causal Video World Model for Real-Time Streaming on Consumer Devices Rajit Rajpal, Shahbuland Matiana, Liew Wei Pyn, Anmol Agarwal, et al. Technical Report (2026) pdf / project page
	Waypoint-1.5: A Real Time Video World Model for Consumer Hardware Rajit Rajpal, Shahbuland Matiana, Liew Wei Pyn, Anmol Agarwal, et al. Technical Report (2026) pdf / project page / huggingface
	Adaptive Momentum and Nonlinear Damping for LLM Pretraining Katerina Karoni, Rajit Rajpal*, Benedict Leimkuhler, Gabriel Stoltz ICML 2026* pdf / code
	Low Rank Friction for Memory-Efficient Transformer Pretraining Rajit Rajpal, Benedict Leimkuhler NeurIPS @ OPT 2026; In Review @ ICLR 2027 pdf / code
	Decoupling Appearance and Geometry in Diffusion World Models Rajit Rajpal, Tadgh Kelly, Maxime Vandegar Preprint (2025) pdf / code
	Adaptive Stepsizing for Stochastic Gradient Langevin Dynamics in Bayesian Neural Networks Rajit Rajpal, Benedict Leimkuhler, Yuanhao Jiang AAAI @ MATH4AI 2025 (Oral, Top 10%) pdf / code
	Overconfidence and the Optimality Trap in Inverse Dynamics Models for Physical AI Rajit Rajpal, Matteo P. A. Preprint (2025)
	Frugal Zero-Shot Diffusion-Based Image Restoration Rajit Rajpal, Marcos Obando, Dolly Chen, Bernardin Amogou Runner-up @ Diffusion Models PhD Hackathon (2024) pdf / code
	Does Translation Invariance Improve Tractable Generative Models? Rajit Rajpal MSc Dissertation (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 Jan 2026 - Present Founding Research Scientist
	Lit8 Apr – Dec 2025 Research Scientist, Video Generation
	Rhombus Power – NASA Ames Research Center Mar 2022 – Mar 2023 Data Scientist

Cloned from here!
Last updated: Aug 11, 2026