Sr. Software Engineer- Backend

As part of the AI Platform team at Databricks, you'll build the substrate that powers data apps, AI agents, model training, model serving, and vector search.

You will:

• Build infrastructure for flagship offerings like MLflow, AI Gateway, Databricks Apps, Agent Framework, Agent Bricks, and Foundation Model APIs
• Improve reliability, latency, and efficiency of distributed AI workloads
• Collaborate with platform, infra, and ML teams to deliver seamless end-to-end experiences
• Shape how developers and data scientists build and interact with AI on Databricks

Requirements:

• 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

Bonus points for:

• Experience with real-time serving, ML infrastructure, or GPU orchestration
• Exposure to platforms like SageMaker, Vertex AI, or Azure ML
• Contributions to OSS projects like MLflow, PyTorch, or Ray
• Built developer platforms or internal tools supporting AI workflows

The pay range for this role is $165,300-$219,675 USD.