Principal Research Scientist - AI Scaling & Optimization

Tech Stack

Responsibilities

• Lead and grow a multidisciplinary research team focused on LLM scaling, efficiency, and systems performance.
• Define the scaling research roadmap in alignment with Databricks’ strategic objectives.
• Drive algorithmic innovations for large-scale neural network training and inference.
• Optimize end-to-end ML systems for distributed training and RL, memory efficiency, and compute efficiency.
• Partner with product and engineering to translate research breakthroughs into customer-impacting capabilities.

Benefits

• Equity

Culture

Requirements