Staff Backend Software Engineer- (AI Platform)
Core
Build the substrate powering data apps, AI agents, model training, model serving, and vector search for the AI Platform team.
Role type
Staff Backend Software Engineer (AI Infrastructure)
Builds
Infrastructure for MLflow, AI Gateway, Databricks Apps, Agent Framework, Agent Bricks, and Foundation Model APIs
Domain
AI Infrastructure / Cloud-Native Distributed Systems
Deliverable
production ML models | infrastructure
Required skills
Backend engineering, distributed systems, scalable APIs, cloud-native infrastructure, service-oriented architecture, deployment pipelines, system observability
Preferred skills
Real-time serving, ML infrastructure, GPU orchestration, OSS contributions (MLflow, PyTorch, Ray), developer platforms
Technologies
Scala, Go, Python
Responsibilities
Build infrastructure for flagship AI offerings, improve reliability/latency/efficiency of distributed AI workloads, collaborate with platform/infra/ML teams, shape developer experiences for AI on Databricks
Seniority
Staff, high-agency IC