Staff Software Engineer - AI Research Infrastructure
Databricks
New York, San Francisco
Workplace: OnsiteFull timeUSD 199,000 - 270,000 annuallyFunction: Software EngineeringExperience: 5+ yearsSkills: ["Distributed systems","GPU","Kubernetes","Slurm","Ray","Rust","C++","Go","Java","Scala","Data pipelines","Model training","Inference workflows"]Lead design and development of the AI research infrastructure powering Databricks AI Research. Build scalable services to schedule, orchestrate and observe large-scale GPU workloads, improve dev tooling, and ensure researchers can iterate quickly without sacrificing reliability, efficiency, or security. Partner with researchers, ML engineers, and platform teams to turn experimental workloads into robust pipelines and push the limits of our infrastructure.
