Databricks
Staff Software Engineer - AI Research Infrastructure
$199K–$270K
New York City, New York +2 · Staff · Full-time · Posted May 15, 2026
- Location
- New York City, New York
San Francisco, California
San Francisco, California - Experience
- 5+ years of software engineering experience, including substantial time working on large-scale distributed systems or infrastructure.
- Education
- BS/MS or PhD in Computer Science or related field.
About the job
Develop and operate the research infrastructure that schedules, orchestrates, and monitors large-scale AI training and inference workloads across thousands of GPUs.
What you'll do
- Build infrastructure for large-scale experiments, data processing, and model training, including job submission, scheduling, and monitoring tools.
- Improve research developer productivity with experiment management, CI/testing infrastructure, and reliable, repeatable workflows.
- Influence the research-compute roadmap and mentor engineers working on compute, infrastructure, and AI systems.
What they ask for
- BS/MS or PhD in Computer Science or a related field.
- Deep experience building and operating distributed systems, data pipelines, or large-scale backend services.
- Proficiency in one or more systems programming languages and experience building or significantly contributing to cluster schedulers, resource managers, or large-scale job orchestration systems.
- Understanding of modern ML training and inference workflows.
Pay
$199,000 — $270,000 USD (base salary)
Tools
C++, Rust, Go, Java, Scala, Kubernetes, Slurm, Ray
AI focus
Build infrastructure and workflows for AI research, including large-scale model training, inference, evaluation, and experimentation.
Model-written summary · the original posting is authoritative