Staff Software Engineer- Foundation Model Inference
Databricks2 months ago
Base Salary
$190k - $265k/yr
Responsibilities
- Build LLM infrastructure for large-scale inference workloads using partner and self-hosted models.
- Improve the reliability, latency, and efficiency of distributed AI workloads.
- Collaborate with platform, infrastructure, and ML teams on end-to-end AI experiences.
- Shape how developers and data scientists build and interact with AI on Databricks.
Requirements
- 8+ years of backend or infrastructure engineering experience.
- Experience with distributed systems, scalable APIs, or cloud-native infrastructure.
- Experience with real-time serving, ML infrastructure, or GPU orchestration.
- Familiarity with service-oriented architecture, deployment pipelines, and system observability.
- Exposure to SageMaker, Vertex AI, or Azure ML is a bonus.
- Contributions to open-source projects such as MLflow, PyTorch, Ray, vLLM, or SGLang are a bonus.
- Experience building developer platforms or internal tools supporting AI workflows is a bonus.
Benefits
- Comprehensive employee 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).