Staff Software Engineer - AI Research Infrastructure

As a Staff Software Engineer, AI Research Infrastructure, you will develop and run the research stack that powers Databricks AI Research. You will design and build services that schedule, orchestrate, and observe large-scale training and inference experiment workloads across thousands of GPUs, improve dev tooling, and ensure researchers can iterate quickly without sacrificing reliability, efficiency, or security. You'll partner closely with research scientists, ML engineers, and platform teams.

BS/MS or PhD in Computer Science or related field; 5+ years of software engineering experience with large-scale distributed systems or infrastructure; deep experience with distributed systems, data pipelines, or large-scale backend services involving GPUs, clusters, or cloud providers; proficiency in systems programming languages (C++, Rust, Go, Java, Scala); contributions to cluster schedulers or job orchestration systems (Kubernetes, Slurm, Ray); understanding of modern ML training and inferer