Staff Software Engineer, AI Runtime: GPU Training Systems

Databricks is seeking a Staff Software Engineer to work on AI Runtime—the AI platform powering large-scale training across domains. You will drive architecture, reliability, and developer experience for multi-node GPU jobs, coordinating with research, product, and customers. You’ll lead critical bets for future GPU generations, optimize performance, and mentor engineers to raise engineering excellence. This role is based in San Francisco with Databricks’ multi-office footprint.