Staff Backend Software Engineer- (AI Platform)
Databricks5 months ago
Base Salary
$166k - $225k/yr
Responsibilities
- Build infrastructure powering MLflow, AI Gateway, Databricks Apps, Agent Framework, Agent Bricks, Foundation Model APIs, and related AI offerings.
- Improve the reliability, latency, and efficiency of distributed AI workloads.
- Collaborate with platform, infrastructure, and machine learning teams to deliver end-to-end experiences.
- Shape how developers and data scientists build and interact with AI on Databricks.
Requirements
- At least 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.
- Experience with real-time serving, ML infrastructure, GPU orchestration, SageMaker, Vertex AI, or Azure ML is a bonus.
- Contributions to open-source projects such as MLflow, PyTorch, or Ray and experience building developer platforms or internal AI workflow tools are bonuses.
Benefits
- Comprehensive benefits and perks, with details varying by region.
Tech Stack
Categories
About Databricks
Databricks is the Data and AI company. More than 20,000 organizations worldwide — including adidas, AT&T, Bayer, Block, Mastercard, Rivian, Unilever, and over 60% of the Fortune 500 — rely on Databricks to build and scale data and AI apps, analytics and agents. Headquartered in San Francisco with 30+ offices around the globe, Databricks offers a unified Data Intelligence Platform that includes Agent Bricks, Lakeflow, Lakehouse, Lakebase and Unity Catalog. --- Databricks applicants Please apply through our official Careers page at databricks.com/company/careers. All official communication from Databricks will come from email addresses ending with @databricks.com or @goodtime.io (our meeting tool).