# Staff Backend Software Engineer- (AI Platform)

**Company:** [Databricks](https://hotfix.jobs/companies/databricks)
**Location:** San Francisco, CA
**Role:** Backend Engineering
**Salary:** $166k – $225k/yr
**Experience:** 5+ years
**Skills:** Scala, Go, Python, Distributed Systems, Scalable Apis, Cloud-Native Infrastructure, Service-Oriented Architecture, MLflow, ML Infrastructure, Gpu Orchestration
**Posted:** 2026-03-31

> Builds infrastructure powering Databricks' AI platform including MLflow, AI Gateway, and model serving. Requires 5+ years backend experience with Scala, Go, or Python, and expertise in distributed systems and scalable APIs.

## Job Description

## Impact/Responsibilities
- Build infrastructure that powers 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.

## Nice-to-Haves
- 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.

## Similar roles

- [Staff Software Engineer, Platform](https://hotfix.jobs/jobs/b2d42759-cf0d-4deb-ac34-f042721e13b6) - Zus Health - Remote - $170k – $200k/yr
- [Staff Software Engineer](https://hotfix.jobs/jobs/150755dc-a13b-49ec-85dd-37f497cecc48) - Chainguard - Remote - $170k – $231k/yr
- [Sr. Member of Technical Staff, Architecture](https://hotfix.jobs/jobs/7d41337d-515a-4461-a422-d394c2791d03) - Illumio - Sunnyvale, CA - $170k – $196k/yr
- [Staff Backend Engineer](https://hotfix.jobs/jobs/5643c913-f735-4ebc-a2d0-78709b88d312) - Sardine - Remote - $170k – $600k/yr
- [Software Engineer - Member of Technical Staff (Platform)](https://hotfix.jobs/jobs/b7d74bf8-8ea7-459c-9f46-6e01d38ca56d) - Mithril - $170k – $230k/yr

**Apply:** https://hotfix.jobs/jobs/0bdcf6ee-395e-4d73-bfcb-e3a85035a930
**Canonical:** https://hotfix.jobs/jobs/0bdcf6ee-395e-4d73-bfcb-e3a85035a930