Staff Backend Software Engineer

At Databricks, we are passionate about enabling data and AI teams to solve the world's toughest problems , from making the next mode of transportation a reality to accelerating the development of medical breakthroughs.

As part of the AI Platform team, you’ll build the substrate that powers everything from data apps, AI agents, model training, model serving, and Vector Search.

The impact you will have:

• Build LLM infrastructure powering trillions of tokens per week for customers through partner models (OpenAI, Anthropic, Gemini) and self-hosted models (Qwen, GPT-OSS, Llama)
• 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

What we look for:

• 8+ 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 , you care about building the right solution, not just any solution

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, Ray, vLLM, SGLang
• Built developer platforms or internal tools supporting AI workflows

The pay range for this role is $190,000-$261,250 USD.

Benefits

At Databricks, we strive to provide comprehensive benefits and perks that meet the needs of all of our employees.