Quick overview
Salary$200k - $400k/yr
SeniorityMid Senior
Work modeOn Site
LocationSunnyvale, United States
Posted7 months ago
RustGitC++PyTorch
Job description
About the Institute of Foundation Models
The Institute of Foundation Models (IFM) designs and operates ultra-scale GPU supercomputing systems to train next-generation foundation models. We believe performance, fault tolerance, and scalability are co-designed across model architecture, communication systems, runtime, and hardware topology.
This role sits at the core of that effort — driving communication performance, distributed reliability, and cross-layer optimization for large-scale training workloads.
The Mission
We are looking for a deeply technical engineer to co-design and optimize the communication stack for large-scale distributed training, including hybrid parallelism and Mixture-of-Experts (MoE) workloads.
This is not a network operations role. This is a systems-level engineering position focused on performance engineering, distributed debugging, and communication-runtime co-design.
· Design and optimize expert-parallel and hybrid-parallel communication patterns
· Drive high-performance hierarchical collectives for MoE workloads
· Co-design runtime orchestration with communication topology awareness
· Reduce tail latency and improve determinism across thousands of GPUs
· Architect fault-tolerant distributed execution under real-world cluster failures
Core Technical Scope
· Communication-compute overlap and topology-aware collective optimization
· Deep debugging of NCCL, RDMA, and custom communication layers
· Hybrid expert parallel strategies in modern large-scale MoE systems
· Elastic and resilient distributed job orchestration concepts
· Congestion analysis and routing optimization across InfiniBand/RoCE fabrics
· Microbenchmarking and performance modeling for communication-heavy workloads
Expected Technical Depth
· Hybrid expert parallel communication for Mixture-of-Experts training
· Scaling behavior under network pressure
· Distributed orchestration for elastic, large-scale training
· Fault detection and recovery in distributed GPU workloads
· Cross-layer bottlenecks: GPU ↔ NIC ↔ PCIe ↔ NVSwitch ↔ Fabric ↔ Scheduler
Required Background
· Experience optimizing distributed training at 1,000+ GPU scale (or equivalent depth)
· Hands-on expertise with RDMA, InfiniBand, RoCE, and GPUDirect RDMA
· Deep familiarity with NCCL and/or UCX internals
· Strong systems programming ability (C/C++, Rust, or Go)
· Strong familiarity with modern model training frameworks such as PyTorch
· Ability to troubleshoot and profile training performance issues related to communication bottlenecks
· Ability to translate research ideas into production-grade optimizations
· Experience debugging distributed hangs, desynchronization, and performance regressions
What We Mean by "Hardcore"
· You can explain why an communication degrades at scale and how to fix it
· You have improved real cluster throughput via communication redesign
· You can trace a distributed hang across ranks and identify the root cause
· You are comfortable working at the boundary between hardware and runtime
Application Requirements
· Include a link to your GitHub (required)
· Provide links to relevant distributed systems, HPC, or large-scale training projects
· Include a list of publications and/or public technical reports (if applicable)
· Describe the hardest distributed debugging problem you solved
· Include measurable performance improvements you have delivered
Academic Qualifications
Master’s, or Bachelor’s + 1 year of relevant experience.
Similar jobs
- CP
Senior .NET Application Developer - Enterprise SystemsNew Auto Apply
AG Consulting Partners, Inc. · Redmond, Washington · Remote 🇺🇸
WorkdaySQLT-SQL+19Technology$95k - $135k/yr9 hours ago - TA
Sales Application Engineer - RES (EE26124)New Auto Apply
TMEIC Corporation Americas · Houston, Texas · Remote 🇺🇸
Microsoft ExcelContinuous ImprovementTechnologySalary not listed13 hours ago - CL

Founding Engineer Auto Apply
Clera · San Francisco · On-site 🇺🇸
SnowflakePythonReact Native+4EngineeringSalary not listed4 months ago - VS

Senior Flight Software EngineerNew Auto Apply
Varda Space Industries · El Segundo · Hybrid 🇺🇸
FiberRustEmbedded Systems+3Technology$169.1k - $216.0k/yr2 days ago - VS

Senior Embedded Software EngineerNew Auto Apply
Varda Space Industries · El Segundo · Hybrid 🇺🇸
FiberRustC+++2Technology$169.1k - $216.0k/yr2 days ago - SP

Sr. Software Engineer Auto Apply
SpaceX · Hawthorne · Hybrid 🇺🇸
DockerSQLScala+17Technology$165k - $230k/yr3 weeks ago - SA

Forward Deployed Engineer I (Remote, US)New Auto Apply
Sayari · Remote - US · Remote 🇺🇸
DockerGCPSQL+14Engineering$160k - $180k/yr2 days ago - CL

Software Engineer Auto Apply
Clera · San Francisco · On-site 🇺🇸
TechnologySalary not listed4 days ago - AI

Senior Software Engineer- Machine LearningNew Auto Apply
Anduril Industries · Broomfield · Hybrid 🇺🇸
Talent AcquisitionFastAPIRust+16TechnologySalary not listedYesterday - AI

Software Engineer- InfrastructureNew Auto Apply
Anduril Industries · Waltham · Hybrid 🇺🇸
Talent AcquisitionDockerRust+19TechnologySalary not listedYesterday - AI

Senior Software Engineer- InfrastructureNew Auto Apply
Anduril Industries · Broomfield · Hybrid 🇺🇸
Talent AcquisitionDockerRust+19Technology$144k - $192k/yrYesterday - ZO

Senior Staff/Principal Software Engineer - GPU System Software, Onboard Execution Auto Apply
Zoox · Foster City · Hybrid 🇺🇸
Technology$339k - $455k/yr3 days ago