Senior Staff Software Engineer, AI RuntimeActive$229K
The opportunity
At Databricks, we run the world's largest AI platform. We believe the next breakthroughs will come from models trained on domain-specific enterprise data, not from bigger general-purpose models alone.
What you'll do
Drive AI Runtime's architecture and lead Databricks' long-term technical: direction in AI training infrastructure.
Define the reliability and observability strategy for a fleet that keeps growing in scale and complexity.
Shape AI Runtime's developer experience for people and agents, so even the: largest training jobs feel as simple as a first experiment.
Anticipate industry trends and choose the technical bets that prepare AI: Runtime for new GPU generations and workload types.
Take on the hardest multi-node training problems yourself, from orchestration: and GPU scheduling to data loading and checkpointing.
Make training faster and cheaper by raising efficiency across model architectures and GPU accelerators.
What they're looking for
- Work with research, product, and customers to decide what we build next and: align teams on hard technical calls.
- Champion engineering excellence, raise the bar, and help us hire and grow great engineers.
- + years in large-scale distributed systems, with significant depth in GPU: training, high-performance computing, or ML systems.
- Hands-on experience training large models with FSDP, DeepSpeed, or Megatron,: and fluency in parallelism strategies.