Staff Software Engineer - Ai Research Infrastructure

Databricks

New York City, New York, US
Base: $190,000 - $270,000 usd; bonus/equity: eligi...
On-site
5+ years software engineering experience
Distributed systems architecture
Gpu cluster and cloud infrastructure
Databricks is seeking a Staff Software Engineer for their AI Research Infrastructure team to develop and optimize large-scale systems for AI model training and experimentation. The ideal candidate will have extensive experience in distributed systems and a strong background in software engineering, particularly with GPUs and cloud infrastructure

Job Summary

  • The role involves designing and building services that schedule and orchestrate large-scale training and inference workloads across thousands of GPUs.
  • Candidates will partner closely with research scientists to turn experimental workloads into robust, repeatable pipelines.
  • The position offers a competitive salary range of $190,000 to $270,000 USD along with eligibility for annual performance bonuses and equity.

Matching Summary

Match Score: 85

Databricks is seeking a Staff Software Engineer for their AI Research Infrastructure team to develop and optimize large-scale systems for AI model training and experimentation. The ideal candidate will have extensive experience in distributed systems and a strong background in software engineering, particularly with GPUs and cloud infrastructure.

Salary

Base: $190,000 - $270,000 USD; Bonus/Equity: Eligible for annual performance bonus and equity; Benefits: Comprehensive benefits package offered

Skills & Requirements

Must-have

  • 5+ years software engineering experience
  • Distributed systems architecture
  • GPU cluster and cloud infrastructure
  • Job scheduling and orchestration systems
  • Systems programming languages like Go or Rust

Nice-to-have

  • Deep understanding of ML training workflows
  • Experience with Kubernetes or Slurm
  • Mentoring junior engineers
  • Translating research needs to infrastructure

Key Requirements

  • BS/MS or PhD in Computer Science
  • 5+ years of software engineering experience
  • Proficiency in C++, Rust, Go, Java, or Scala

Work Rights

Not specified

Tailored Resume

Cover Letter