Staff Backend Software Engineer- (AI Platform)
Databricks6 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 builds a cloud-based data and AI platform centered on the lakehouse architecture, combining data engineering, analytics, and machine learning with Apache Spark, Delta Lake, and MLflow. It sells subscriptions and cloud services to enterprises that need to unify data pipelines and develop large-scale AI and analytics. Founded in 2013 by the creators of Apache Spark and headquartered in San Francisco, the company is privately held and serves organizations across many industries.