Staff Backend Engineer — AI Infrastructure & Platform

Databricks Inc.; San Francisco (CA); On-site; USD 166,000 - 225,000; Full time

Job summary

Databricks, a leading Data + AI company, is seeking a Staff Backend Software Engineer for its AI Platform. You will contribute to foundational infrastructure powering MLflow, AI gateway, Agent Framework, and Foundation Model APIs, while improving reliability, latency, and efficiency of distributed workloads.

You will collaborate across platform, infra, and ML teams to deliver seamless experiences for developers and data scientists.

Qualifications

• 5+ years of backend or infrastructure engineering experience.
• Strong programming skills in Scala, Go, or Python.
• Familiarity with distributed systems and cloud-native infrastructure.
• Experience with service-oriented architectures and deployment pipelines.
• Ownership mindset and focus on building the right solution.

Responsibilities

• Build infrastructure powering Databricks AI offerings like MLflow and Foundation Model APIs.
• Improve reliability, latency, and efficiency of distributed AI workloads.
• Collaborate with platform, infra, and ML teams for end-to-end experiences.
• Shape how developers and data scientists build and interact with AI on Databricks.

Skills

Scala

Go

Python

Job description

Databricks, a leading Data + AI company, is seeking a Staff Backend Software Engineer for its AI Platform. You will contribute to foundational infrastructure powering MLflow, AI gateway, Agent Framework, and Foundation Model APIs, while improving reliability, latency, and efficiency of distributed workloads.

You will collaborate across platform, infra, and ML teams to deliver seamless experiences for developers and data scientists.