Staff Software Engineer - AI Research Infrastructure

Highlights

• Build research stack powering Databricks AI across thousands of GPUs.
• Turn experimental workloads into robust, repeatable pipelines for models.
• Shape roadmap for how AI Research trains, evaluates and ships models.

What they offer

• Annual performance bonus
• Equity
• Comprehensive benefits

Ideal candidate

Seasoned infra builder with 5+ years scaling distributed systems who thrives translating research needs into reliable platforms. Pragmatic, fast-moving mentor passionate about operational excellence and accelerating AI discovery.

Requirements

• BS/MS or PhD in Computer Science or related field
• 5+ years of software engineering experience including large-scale distributed systems or infrastructure
• Deep experience building and operating distributed systems, data pipelines, or large-scale backend services
• Proficient in systems programming languages such as C++, Rust, Go, Java, or Scala
• Built or contributed to cluster schedulers, resource managers, or job orchestration systems
• Understanding of modern ML training and inference workflows including distributed training and fine-tuning
• Track record driving complex systems from prototype to stable, well-owned services
• Clear communication across researchers and engineers with operational excellence mindset