Databricks01.05.2026

Staff Software Engineer - AI Research Infrastructure

New York City

Обязанности

01Design and implement infrastructure that supports large-scale experiments, data processing, and model training
02Enable researchers to go from idea to large-scale experiment in minutes, not days, by building powerful abstractions for job submission, scheduling, and monitoring
03Create tooling that improves research developer productivity, such as experiment management systems, CI/testing infrastructure for research code, and workflows that reduce iteration time
04Influence the long-term roadmap for research computation, shaping how Databricks AI Research train, evaluate, and ship models to customers
05Serve as a technical mentor and force multiplier for other engineers working on compute, infra, and AI systems

01BS/MS or PhD in Computer Science or related field
025+ years of software engineering experience, including substantial time working on large-scale distributed systems or infrastructure
03Deep experience with building and operating distributed systems, data pipelines, or large-scale backend services, ideally involving GPUs, clusters, or major cloud providers
04Proficient in one or more systems programming languages (e.g., C++, Rust, Go, Java, Scala) and can design, implement, and debug complex services
05Have built or significantly contributed to cluster schedulers, resource managers, or large-scale job orchestration systems (e.g., Kubernetes, Slurm, Ray, custom internal systems)
06Understand modern ML training and inference workflows (e.g., distributed training, model parallelism, fine-tuning, evaluation)
07Can move fast and be pragmatic in getting things done, while caring about operational excellence
08Have driven complex systems from prototype to stable, well-owned services
09Communicate clearly with both researchers and engineers, and enjoy translating between research needs and infra realities