As a Staff Software Engineer in AI Research Infrastructure at Databricks, you will develop and manage the research stack that supports AI model development. Your role will involve designing services for large-scale training and inference workloads across numerous GPUs. You will work closely with research scientists and ML engineers to create robust pipelines and improve developer productivity. This position emphasizes operational excellence while enabling rapid experimentation and iteration.

• BS/MS or PhD in Computer Science or a related field
• 5+ years of software engineering experience, particularly with large-scale distributed systems or infrastructure
• Deep experience in building and operating distributed systems, data pipelines, or large-scale backend services
• Proficiency in one or more systems programming languages such as C++, Rust, Go, Java, or Scala
• Experience with cluster schedulers, resource managers, or large-scale job orchestration systems
• Understanding of modern ML training and inference workflows
• Ability to communicate effectively with both researchers and engineers

• Design and implement infrastructure that supports large-scale experiments and model training
• Enable researchers to conduct large-scale experiments quickly by building powerful job submission and monitoring tools
• Create tooling that enhances research developer productivity, including experiment management systems and CI/testing infrastructure
• Influence the long-term roadmap for research computation and model deployment
• Serve as a technical mentor for other engineers working on compute and AI systems