Principal Research Scientist – Scaling
Core
Lead a multidisciplinary research team to advance the state of the art in large-scale machine learning, focusing on post-training, RL, and inference efficiency for LLMs.
Role type
Principal Research Scientist (Team Lead)
Builds
Databricks AI platform capabilities for training, serving, and adapting LLMs at scale
Domain
Artificial Intelligence / Large Language Models / Distributed Systems
Deliverable
production ML models
Required skills
Foundation model efficiency, LLM scaling, distributed ML systems, model optimization, Python, PyTorch, team leadership, research roadmap definition, algorithmic innovation, system optimization, responsible AI
Preferred skills
Systems and ML intersection, distributed training frameworks, compiler/kernel optimization, top-tier conference publications, open-source contributions
Technologies
Python, PyTorch
Responsibilities
Lead and grow a multidisciplinary research team; Define the scaling research roadmap; Drive algorithmic innovations for large-scale neural network training and inference; Optimize end-to-end ML systems for distributed training and RL; Partner with product and engineering to translate research into platform capabilities; Mentor and develop talent; Represent Databricks AI research externally through publications and talks
Seniority
Principal, hands-on IC with team leadership