Staff Software Engineer - AI Research Infrastructure
Databricks - San Francisco, CA
Hiring: Staff Software Engineer - AI Research Infrastructure Company: Databricks Location: San Francisco, CA Job Posted Time: 2026-09-10 11:36:55 Target Skills & Keywords : C++, Data Pipeline, Databricks, Java, Kubernetes, LLM, Microsoft Excel, Rust, Scala About the job Experience: •5+ years of software engineering experience, including substantial time working on large‑scale distributed systems or infrastructure. Required Skills: •You’ll partner closely with research scientists, ML engineers, and platform teams to turn experimental workloads into robust, repeatable pipelines, and to push the limits of what our infrastructure can support. •The Impact you will have •Design and implement infrastructure that supports large‑scale experiments, data processing, and model training (e.g., HPC clusters, GPU fleets, or cloud‑based systems) •Enable researchers to go from idea to large‑scale experiment in minutes, not days, by building powerful abstractions for job submission, scheduling, and monitoring. •Create tooling that improves research developer productivity, such as experiment management systems, CI/testing infrastructure for research code, and workflows that reduce iteration time. •Influence the long‑term roadmap for research computation, shaping how Databricks AI Research train, evaluate, and ship models to customers. •Serve as a technical mentor and force multiplier for other engineers working on compute, infra, and AI systems. •5+ years of software engineering experience, including substantial time working on large‑scale distributed systems or infrastructure. Qualifications: •What We Look For BS/MS or PhD in Computer Science or related field Compensation: •$199,000 - $270,000 / year •Competitive benefits and rewards package Interested candidates, please apply directly through the job posting on company's career page or try via AI auto apply on this platform. Don't miss this opportunity to join a forward-thinking team!