Senior AI Runtime Engineer — GPU Training Platform

Databricks; Mountain View (CA); On-site; USD 160,000 - 225,000; Full time

Benefits offered by this job

Comprehensive benefits

Annual performance bonus

Equity options

Job summary

Databricks is hiring a Senior Software Engineer for AI Runtime in Mountain View, California. In this role, you'll drive the architecture and evolution of AIR's managed GPU training platform, ensuring scalable and resilient training. The ideal candidate has 5+ years of experience with large-scale distributed systems and strong knowledge of GPU performance and training infrastructure.

This position comes with a competitive salary range of $160,000 – $225,000 and a commitment to diversity and inclusion in the workplace.

Qualifications

• 5+ years of experience building and operating large-scale distributed systems.
• Experience with GPU training infrastructure, high-performance computing, or ML systems.
• Strong understanding of training resilience patterns.

Responsibilities

• Drive the architecture and evolution of AIR's managed GPU training platform.
• Solve the hardest problems in large-scale training.
• Push GPU efficiency and training performance.
• Partner with various teams to shape APIs and developer experience.

Skills

Large-scale distributed systems

GPU training infrastructure

High-performance computing

ML systems

Distributed training frameworks

Algorithms

Data structures

System design

Communication skills

Education

BS in Computer Science or related field

MS or PhD preferred

Tools

PyTorch

FSDP

DeepSpeed

Megatron

Job description

Databricks is hiring a Senior Software Engineer for AI Runtime in Mountain View, California. In this role, you'll drive the architecture and evolution of AIR's managed GPU training platform, ensuring scalable and resilient training. The ideal candidate has 5+ years of experience with large-scale distributed systems and strong knowledge of GPU performance and training infrastructure.

This position comes with a competitive salary range of $160,000 – $225,000 and a commitment to diversity and inclusion in the workplace.