# Staff Software Engineer - AI Research Infrastructure

**Company:** [Databricks](https://hotfix.jobs/companies/databricks)
**Location:** New York, NY, San Francisco, CA
**Role:** DevOps / SRE
**Salary:** $199k – $270k/yr
**Experience:** 5+ years
**Skills:** Kubernetes, Slurm, Ray, C++, Rust, Go, Java, Scala, Gpu Clusters, Distributed Systems
**Posted:** 2026-05-15

> Build and operate the large-scale training and inference infrastructure that powers Databricks AI Research, enabling researchers to run experiments across thousands of GPUs. Partner with ML scientists and platform teams to deliver reliable, high-performance orchestration and tooling.

## Job Description

## Responsibilities
- Design and implement infrastructure that supports large‑scale experiments, data processing, and model training (e.g., HPC clusters, GPU fleets, or cloud‑based systems)
- Build powerful abstractions for job submission, scheduling, and monitoring so researchers can go from idea to large‑scale experiment in minutes
- 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 trains, evaluates, and ships models
- Serve as a technical mentor and force multiplier for other engineers working on compute, infra, and AI systems

## Requirements
- BS/MS or PhD in Computer Science or related field
- 5+ years of software engineering experience, including substantial time working on large‑scale distributed systems or infrastructure
- Deep experience building and operating distributed systems, data pipelines, or large‑scale backend services, ideally involving GPUs, clusters, or major cloud providers
- Proficient in one or more systems programming languages (e.g., C++, Rust, Go, Java, Scala) and able to design, implement, and debug complex services
- Experience building or significantly contributing to cluster schedulers, resource managers, or large‑scale job orchestration systems (e.g., Kubernetes, Slurm, Ray, custom internal systems)
- Understanding of modern ML training and inference workflows (e.g., distributed training, model parallelism, fine‑tuning, evaluation)
- Ability to move fast and be pragmatic while caring about operational excellence; experience driving complex systems from prototype to stable, well‑owned services
- Strong communication skills with both researchers and engineers

## Similar roles

- [Staff Site Reliability Engineer, Production Engineering](https://hotfix.jobs/jobs/7459df86-95cc-4cb0-9b82-05f8c00e9a78) - Dropbox - Remote - $199k – $302k/yr
- [Staff Infrastructure Engineer](https://hotfix.jobs/jobs/75f07283-34de-4989-b193-11c3c3676fd2) - Aurelian - Seattle, WA - $200k – $300k/yr
- [Senior / Staff Platform Engineer](https://hotfix.jobs/jobs/1e15e953-b6f5-46a6-afc4-36a6811e809e) - Radar Labs - New York, NY - $200k – $300k/yr
- [Staff Software Engineer, Infrastructure](https://hotfix.jobs/jobs/ecfe4e28-57ed-4e1d-a9fb-5917d180aeec) - F2 - San Francisco, CA - $200k – $300k/yr
- [Member of Technical Staff, DevOps](https://hotfix.jobs/jobs/c99fed69-8e72-438a-9f3d-3e18894667e7) - Vapi - San Francisco, CA - $200k – $270k/yr

**Apply:** https://hotfix.jobs/jobs/1147577e-c726-4cb2-8f5f-c6b601ab19c6
**Canonical:** https://hotfix.jobs/jobs/1147577e-c726-4cb2-8f5f-c6b601ab19c6