The role is part of the AI Platform team at Databricks, focusing on building the infrastructure that supports various AI applications, including model training and serving. The team operates at the forefront of AI infrastructure, collaborating closely with research and product teams. The position emphasizes improving the reliability and efficiency of distributed AI workloads while enhancing the developer experience. Candidates will have the opportunity to work on flagship offerings like MLflow and AI Gateway.

• 5+ years of experience in backend or infrastructure engineering
• Strong programming skills in Scala, Go, or Python
• Experience with distributed systems, scalable APIs, or cloud-native infrastructure
• Familiarity with service-oriented architecture, deployment pipelines, and system observability
• Strong product and ownership mindset

• Build infrastructure that powers flagship offerings like MLflow and AI Gateway
• Improve reliability, latency, and efficiency of distributed AI workloads
• Collaborate with platform, infrastructure, and machine learning teams to deliver seamless end-to-end experiences
• Shape how developers and data scientists build and interact with AI on Databricks