Principal AI Research Scientist, Research Director - AI Scaling

Job summary

Responsibilities

As a Principal Research Scientist – AI Scaling, you will lead a team of world‑class researchers and engineers to advance the state of the art in large‑scale machine learning, focusing on post-training, RL and inference efficiency, optimization, and scaling.

• Lead and grow a multidisciplinary research team focused on foundational and applied AI problems, with a particular emphasis on LLM scaling, efficiency, and systems performance.
• Define the scaling research roadmap in alignment with Databricks’ strategic objectives, prioritizing advances in foundation model efficiency and large‑scale training and inference.
• Drive algorithmic innovations for large‑scale neural network training and inference, including novel optimizers, low‑precision techniques, and model adaptation methods, and guide…
• Optimize end‑to‑end ML systems for distributed training and RL, memory efficiency, and compute efficiency through close collaboration with core systems and platform teams…
• Partner with product and engineering to translate research breakthroughs, especially around scaling and efficiency, into customer‑impacting capabilities in the Databricks AI…

Tech stack

• Python
• PyTorch