Staff Software Engineer - Ai Research Infrastructure

Databricks

New York, United States
Base: $190,000 - $270,000 usd; bonus/equity: eligi...
On-site
5+ years distributed systems experience
Proficiency in c++, rust, go, java, or scala
Experience with kubernetes, slurm, ray, or custom schedulers
The role involves designing and implementing infrastructure that supports large-scale experiments, data processing, and model training across thousands of GPUs

Job Summary

  • The role involves designing and implementing infrastructure that supports large-scale experiments, data processing, and model training across thousands of GPUs.
  • You will partner closely with research scientists and ML engineers to turn experimental workloads into robust, repeatable pipelines.
  • Databricks is committed to fair compensation practices with a salary range of $190,000 to $270,000 USD plus eligibility for annual performance bonus and equity.

Matching Summary

The role involves designing and implementing infrastructure that supports large-scale experiments, data processing, and model training across thousands of GPUs.

Salary

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

Skills & Requirements

Must-have

  • 5+ years distributed systems experience
  • Proficiency in C++, Rust, Go, Java, or Scala
  • Experience with Kubernetes, Slurm, Ray, or custom schedulers
  • Understanding of ML training workflows and model parallelism
  • Building large-scale backend services and data pipelines

Nice-to-have

  • Deep expertise in GPU fleets and HPC clusters
  • Ability to translate research needs into infrastructure solutions
  • Experience mentoring engineers and driving technical roadmaps
  • Pragmatic approach to moving from prototype to stable service

Key Requirements

  • BS/MS or PhD in Computer Science or related field
  • 5+ years of software engineering experience
  • Substantial time working on large-scale distributed systems

Work Rights

Not specified

Tailored Resume

Cover Letter