Mohit Yadav
Lead AI Scientist @ Optum AI • Bridging AI research ⇄ impact in healthcare.
Past affiliations: UMass Amherst, Pythia Labs, Ai2, Amazon, and IIT Mandi.
Sunnyvale, California
San Francisco Bay Area
Hey, thanks for stopping by! 👋
At Optum AI, I develop agentic AI systems for healthcare NLP, building human-in-the-loop solutions to support research planning and report generation through information extraction and efficient tool planning. Lately, I’ve been interested in developing efficient algorithms for LLM inference and reasoning.
Previously, I’ve worked on generative AI for protein design at a bio-tech startup, and completed my Ph.D. in machine learning at the University of Massachusetts Amherst (May 2023), co-advised by Prof. Daniel Sheldon and Prof. Cameron Musco. My research was supported by university fellowships and focused on deep learning and fast linear-algebra methods for Gaussian processes.
I enjoy collaborating on interesting ML problems and discussing how research leads to real-world impact. Feel free to reach out or connect on LinkedIn or email.
Highlights
| Nov 01, 2025 | Joined Optum AI as Lead AI Scientist 🎉! Building AI systems that simplify healthcare complexity. 🚀 |
|---|---|
| Nov 07, 2024 | De Novo antibody design using Generative AI and (discrete) flow matching — Patent Published! |
| Sep 26, 2024 | Paper accepted to NeurIPS 2024! 🎉 “Gaussian process bandits for top-k recommendations” — excited to share this work on scalable bandit algorithms for ranking problems. |
| May 01, 2024 | PhD thesis available! Look at this beauty 🎓 — Fast Linear Algebra for Gaussian Processes |
| Nov 16, 2023 | Successfully defended my PhD thesis! 🎓 Excited to join Pythia Labs to work on generative AI for protein design. 🧬 |
