Staff AI Research Infra Engineer - GPU Scale & Equity

Databricks Inc.; New York (NY); On-site; USD 199,000 - 270,000; Full time

Job summary

Databricks Inc. seeks a Staff Software Engineer - AI Research Infrastructure in New York, NY. You'll develop and manage the infrastructure for AI research, including designing systems for large-scale experiments and model training. With a BS/MS or PhD in Computer Science and 5+ years of software engineering experience, you should excel in building distributed systems and have experience with systems programming languages. A commitment to operational excellence and effective communication with cross-functional teams is essential. The role offers a competitive salary range of $199,000 – $270,000 USD, depending on experience and skills.

Qualifications

• 5+ years of software engineering experience, including substantial time working on large-scale distributed systems.
• Deep experience with building and operating distributed systems, data pipelines, or large-scale backend services.
• Proficient in one or more programming languages (e.g., C++, Rust, Go, Java, Scala).

Responsibilities

• Develop and run the research stack that powers Databricks AI Research.
• Design and implement infrastructure for large-scale experiments and model training.
• Build tooling that improves research developer productivity.

Skills

Large-scale distributed systems

Software engineering

Systems programming languages

Cluster schedulers

Modern ML training and inference workflows

Education

BS/MS or PhD in Computer Science or related field

Tools

Kubernetes

Slurm

Ray

Job description

Databricks Inc. seeks a Staff Software Engineer - AI Research Infrastructure in New York, NY. You'll develop and manage the infrastructure for AI research, including designing systems for large-scale experiments and model training. With a BS/MS or PhD in Computer Science and 5+ years of software engineering experience, you should excel in building distributed systems and have experience with systems programming languages. A commitment to operational excellence and effective communication with cross-functional teams is essential. The role offers a competitive salary range of $199,000 – $270,000 USD, depending on experience and skills.