Staff Backend Software Engineer
Overview
As a member of the AI Platform team at Databricks, you’ll help build the substrate powering AI workloads—from data apps to model training and serving. You will tackle reliability, latency, and efficiency for distributed AI tasks and collaborate across platform, infra, and ML squads to deliver seamless experiences. You’ll influence how developers and data scientists interact with AI on Databricks, contributing to a fast-growing AI infrastructure ecosystem. This role offers the chance to shape scalable, enterprise-grade AI infrastructure at scale, collaborating with partner and self-hosted models.
ResponsibilitiesBuild and improve LLM infrastructure handling trillions of tokens weekly with partner and self-hosted modelsEnhance reliability, latency, and efficiency of distributed AI workloadsCollaborate with platform, infra, and ML teams to deliver end-to-end AI experiencesShape developer and data scientist interactions with AI on Databricks
Key requirements8+ years of backend or infrastructure engineering experienceStrong programming skills in Scala, Go, or PythonExperience with distributed systems, scalable APIs, or cloud-native infrastructureFamiliarity with service-oriented architecture, deployment pipelines, and system observabilityStrong product and ownership mindsetownershipcollaborationproblem-solvingScalaGoPython