Databricks
Staff Backend Software Engineer- (Databricks AI)
$166K–$225K
San Francisco, California +1 · Staff · Full-time · Posted Mar 31, 2026
- Location
- San Francisco, California
Mountain View, California - Experience
- 5+ years of experience in backend or infrastructure engineering.
About the job
Build backend and infrastructure for Databricks’ AI platform, supporting AI products and improving the performance and reliability of distributed AI workloads.
What you'll do
- Build infrastructure for products including MLflow, AI Gateway, Databricks Apps, Agent Framework, Agent Bricks, and Foundation Model APIs.
- Improve the reliability, latency, and efficiency of distributed AI workloads.
- Collaborate with platform, infrastructure, and ML teams to deliver end-to-end experiences and shape how developers and data scientists use AI on Databricks.
What they ask for
- 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.
Nice to have
- Experience with real-time serving, ML infrastructure, or GPU orchestration; exposure to SageMaker, Vertex AI, or Azure ML.
- Contributions to OSS projects such as MLflow, PyTorch, or Ray; experience building developer platforms or internal tools supporting AI workflows.
Pay
$166,000 — $225,000 USD (base salary)
Tools
Scala, Go, Python, SageMaker, Vertex AI, Azure ML, MLflow, PyTorch
AI focus
Build infrastructure for AI agents, model training and serving, vector search, and other AI workloads; support Databricks AI products and improve distributed AI infrastructure.
Model-written summary · the original posting is authoritative