Senior Staff Software Engineer, AI Runtime

📋 Description

• Drive AI Runtime's architecture and lead Databricks' long-term technical direction in AI training
• Define the reliability and observability strategy for a fleet that keeps growing in scale and
• Shape AI Runtime's developer experience for people and agents, so even the largest training jobs
• Anticipate industry trends and choose the technical bets that prepare AI Runtime for new GPU
• Take on the hardest multi-node training problems yourself, from orchestration and GPU scheduling to
• Make training faster and cheaper by raising efficiency across model architectures and GPU

🎯 Requirements

• 15+ years in large-scale distributed systems, with significant depth in GPU training
• Hands-on experience training large models with FSDP, DeepSpeed, or Megatron, and fluency in
• Expertise in GPU performance, including accelerator architecture, high-speed interconnects, and
• Strong understanding of training resilience patterns, such as checkpointing, failure detection, and
• A background in building and running multi-tenant cloud platforms as products.
• A track record of setting technical direction across teams and seeing it through to business impact.

🎁 Benefits

• Comprehensive benefits and perks
• Annual performance bonus
• Equity