Staff Software Engineer, AI Runtime at Databricks · Mountain View, CA · Joined
D
Staff Software Engineer, AI Runtime
Databricks · Mountain View, CA · Just now · 0 applicants
NewOn-siteFull-timeLead
Pay
$190k–$265kListed
Workplace
On-site
Job type
Full-time
Level
Lead
Team: AI Runtime · Source: jobright · Posted by joined-scrapper
About the role
Databricks is seeking a Staff Software Engineer to shape the architecture and technical direction of AI Runtime, a serverless GPU training platform, while improving the reliability, performance, and developer experience of its AI training infrastructure.
What you’ll do
Shape AI Runtime's architecture and technical direction
Improve reliability, performance, and developer experience of AI training infrastructure
Lead technical projects across teams to customer impact
Mentor engineers and drive alignment on technical designs
What you’ll need
10+ years in large-scale distributed systems, with depth in GPU training, HPC, or ML systems
Hands-on large model training with FSDP, DeepSpeed, or Megatron and fluency in parallelism strategies
Expertise in GPU performance, including accelerator architecture, high-speed interconnects, and kernels
Strong understanding of training resilience patterns such as checkpointing, failure detection, and recovery
Experience building and running multi-tenant cloud platforms as products; BS in Computer Science or related field
Track record leading technical projects across teams and mentoring engineers
Original job description
Databricks builds AI Runtime, a serverless product for GPU training of models using enterprise data. The Staff Software Engineer will shape AI Runtime’s architecture and technical direction while improving the reliability, performance, and developer experience of its AI training infrastructure.10+ years in large-scale distributed systems, with significant…