Staff Software Engineer - AI Research Infrastructure

As a Staff Software Engineer, AI Research Infrastructure, you will design and build services that schedule, orchestrate, and observe large-scale training and inference experiment workloads across thousands of GPUs, and partner closely with research scientists, ML engineers, and platform teams to turn experimental workloads into robust, repeatable pipelines.

Requirements

• BS/MS or PhD in Computer Science or related field
• 5+ years of software engineering experience, including substantial time working on large-scale distributed systems or infrastructure
• Have deep experience with building and operating distributed systems, data pipelines, or large-scale backend services, ideally involving GPUs, clusters, or major cloud providers
• Are proficient in one or more systems programming languages (e.g., C++, Rust, Go, Java, Scala) and can design, implement, and debug complex services
• Have built or significantly contributed to cluster schedulers, resource managers, or large-scale job orchestration systems (e.g., Kubernetes, Slurm, Ray, custom internal systems)
• Understand modern ML training and inference workflows (e.g., distributed training, model parallelism, fine-tuning, evaluation)
• Can move fast and be pragmatic in getting things done, while caring about operational excellence

Benefits

• Generous Paid Time Off
• 401k Matching
• Retirement Plan
• Relocation Assistance