Druv Pai

MoTS @ Thinking Machines, PhD Student @ BAIR

San Francisco Bay Area

About

Experience

  • Member of Technical Staff at Thinking Machines Lab
    Mar 2026 - Present · 4 mos

    Research

  • Student Researcher at Berkeley Artificial Intelligence Research
    Jan 2021 - Present · 5 yrs 6 mos

    Advisors: Prof. Yi Ma, Prof. Jiantao Jiao, and Prof. Jason Lee.

  • Student Researcher at Google DeepMind
    May 2025 - Dec 2025 · 8 mos

    - Worked on science of synthetic data and self-improvement in continual pre-training (Summer 2025, Google Research) - Worked on next-generation foundation model architectures (Fall 2025, Google DeepMind)

  • Machine Learning Engineer Intern at Nexusflow
    May 2024 - Dec 2024 · 8 mos

    Worked on LLMs for function calling and agent flows, specifically involved in overall function calling methodology, data curation pipelines, evaluation development, and infrastructure. Helped produce state-of-the-art open-weights model family Athene-V2.

  • Course Staff at UC Berkeley College of Engineering
    Jan 2020 - Dec 2024 · 5 yrs

    Course staff (including Head TA positions) for EE226A, EE127/227A, EE16B, and CS 170.