Principal Research Scientist - Ai Scaling & Optimization

Databricks

Mountain View, California, US
Base: $270,000 - $350,000 usd; bonus/equity: eligi...
On-site
Deep expertise in llms and distributed ml systems
Hands-on leadership with python and pytorch
Proven ability to lead research teams
Databricks is seeking a Principal Research Scientist specializing in AI Scaling & Optimization to lead a team focused on enhancing large-scale machine learning efficiencies. The ideal candidate will have a strong background in foundational AI models, deep learning, and experience in translating research into scalable product capabilities

Job Summary

  • The role involves leading a team of researchers to advance the state of the art in large-scale machine learning with a focus on post-training, RL, and inference efficiency.
  • Candidates must define and execute a research roadmap that delivers tangible improvements to how customers train, serve, and adapt LLMs at scale.
  • The position requires close collaboration with product and engineering leaders to translate cutting-edge methods into performant, reliable infrastructure.

Matching Summary

Match Score: 85

Databricks is seeking a Principal Research Scientist specializing in AI Scaling & Optimization to lead a team focused on enhancing large-scale machine learning efficiencies. The ideal candidate will have a strong background in foundational AI models, deep learning, and experience in translating research into scalable product capabilities.

Salary

Base: $270,000 - $350,000 USD; Bonus/Equity: Eligible for annual performance bonus and equity; Benefits: Not specified

Skills & Requirements

Must-have

  • Deep expertise in LLMs and distributed ML systems
  • Hands-on leadership with Python and PyTorch
  • Proven ability to lead research teams
  • Translate research into scalable product capabilities

Nice-to-have

  • Strong industry and academic network in large-scale ML
  • First-author publications at top ML/systems conferences
  • Experience with compiler and kernel optimization

Key Requirements

  • Proven track record of industry impact in foundation model efficiency
  • Strong programming skills in Python and PyTorch
  • Demonstrated ability to influence cross-functional roadmaps

Work Rights

Not specified

Tailored Resume

Cover Letter