Training Infrastructure jobs
Training infrastructure teams build the machinery under the runs: distributed frameworks, checkpointing that survives node failures, data loaders that keep thousands of GPUs from starving, and the observability to explain a slow step time. At sufficient scale hardware fails daily, which makes reliability engineering its own discipline here.
The vocabulary is FSDP, Megatron, DeepSpeed, and JAX's sharding machinery, plus storage tuned for sequential reads at absurd throughput. Debugging spans the whole stack, from NCCL timeouts to a flaky top-of-rack switch. Labs hire this continuously, and strong systems engineers without ML backgrounds cross over here regularly.
137open roles right now
- Senior/Staff Engineer : Post Silicon- Bring UpCerebrasOn-site · Staff+ · Infrastructure$175k–$275k
- Lead RTL Design EngineerCerebrasHybrid · Lead / Manager · Infrastructure$175k–$275k
- Research Engineer, ML InfrastructureCognitionOn-site · Senior · Infrastructure
- Senior Hardware Test and Reliability EngineerSkydioOn-site · Senior · Infrastructure$170k–$230k
- Staff Software Engineer - Managed KubernetesLambdaHybrid · Staff+ · Infrastructure$314k–$465k
- Principal Engineer, Embedded Vehicle SoftwareWayveHybrid · Staff+ · Infrastructure$348k–$458k
- Hardware Test & Reliability InternSkydioOn-site · Intern · Infrastructure$47–$58/hr
- Research Engineer, Code Agents InfraMistral AI · Senior · Infrastructure
- Industrial ComputeOpenAIRemote (US) · Infrastructure$150k–$300k
- Staff Software Engineer - StreamingDatabricksRemote (US) · Staff+ · Infrastructure$182k–$247k
- Member of Technical Staff, AI Training InfrastructureFireworks AIOn-site · Mid-level · Infrastructure$200k–$350k
- Senior/Staff Software Engineer, AI Agent InfrastructureNuroOn-site · Staff+ · Infrastructure$194k–$352k
- Embedded Software Engineer, Onboard Test AutomationWayveHybrid · Mid-level · Infrastructure$210k–$267k
- Staff Software Engineer, Environments InfrastructureAnthropicHybrid · Staff+ · Infrastructure$405k–$605k
- Robotics EngineerLuma AIOn-site · Senior · Infrastructure
- Research Scientist / Engineer – Training InfrastructureLuma AIOn-site · Senior · Infrastructure
- Test Validation EngineerNuroOn-site · Mid-level · Infrastructure$132k–$198k
- Kernel Engineer - New GradCerebrasRemote (US, Canada) · Junior · Infrastructure
- Software Engineer - EvalsxAIOn-site · Mid-level · Infrastructure$175k–$275k
- Simulation Infrastructure EngineerOpenAIOn-site · Senior · Infrastructure$230k–$385k
- ML Infrastructure EngineerxAIOn-site · Mid-level · Infrastructure$180k–$440k
- Machine Learning Infrastructure Engineer, Safeguards ResearchAnthropicHybrid · Mid-level · Infrastructure$350k–$500k
- Software Engineer, Compute Foundations SystemsOpenAIOn-site · Mid-level · Infrastructure$230k–$490k
- Software Engineer, ML Platform (ML Training)ZooxOn-site · Mid-level · Infrastructure$182k–$257k
- Software Engineer - Platform Infrastructure (Rust, C++)xAIOn-site · Mid-level · Infrastructure$180k–$440k
- Staff Software Engineer - Infrastructure StorageLambdaHybrid · Staff+ · Infrastructure$314k–$465k
- Simulation EngineerCerebrasOn-site · Junior · Infrastructure
- Member of Technical Staff - RL Training FrameworkxAIOn-site · Mid-level · Infrastructure$180k–$440k
- Member of Engineering (Agent Sandboxes)PoolsideRemote (EMEA, US) · Mid-level · Infrastructure
- Research Engineer, Data InfrastructureMistral AIRemote (FR, UK, PL, …) · Mid-level · Infrastructure
- Research Software EngineerMistral AIRemote (FR, UK) · Mid-level · Infrastructure
- Research Platform EngineerMistral AIRemote (FR, UK, PL) · Senior · Infrastructure
- Engineering Manager, MLCursor (Anysphere)On-site · Lead / Manager · Infrastructure
- Principal Software Engineer, SimulationOpenAIOn-site · Staff+ · Infrastructure$347k–$490k
- Sr. Engineering Manager, AI RuntimeDatabricksOn-site · Lead / Manager · Infrastructure$229k–$297k
- Systems Test Engineer, System Behavior AnalysisZooxOn-site · Mid-level · Infrastructure$143k–$177k
- Staff Engineer, HMS Factory (R5276)Shield AIOn-site · Staff+ · Infrastructure$150k–$220k
- Head of Systems Engineering (Sr. Director)NuroOn-site · Staff+ · Infrastructure$284k–$426k
- Senior Software Engineer II, Applied TrainingCoreWeaveOn-site · Senior · Infrastructure$182k–$242k
- Systems Test Engineer (R5243)Shield AIOn-site · Senior · Infrastructure
- Senior/Staff Software Engineer - C++ Simulation PlatformZooxOn-site · Staff+ · Infrastructure$245k–$305k
- Prototyping Lab Technician, RoboticsOpenAIOn-site · Junior · Infrastructure$225k–$295k
- Principal Engineer, Data & ComputeWayveHybrid · Staff+ · Infrastructure$370k–$418k
- Engineering Site LeadPerplexityHybrid · Lead / Manager · Infrastructure
- Senior Software Engineer, Sensor PlatformNuroOn-site · Senior · Infrastructure$160k–$241k
- Senior Software Engineer, ML InfrastructureNuroOn-site · Senior · Infrastructure$194k–$291k
- Senior/Staff Software Engineer - Planner Frameworks PipelineZooxOn-site · Staff+ · Infrastructure$219k–$315k
- Member of Technical StaffFireworks AI · Senior · Infrastructure$175k–$220k
- Software Engineer, Onboard InfrastructureNuroOn-site · Mid-level · Infrastructure$160k–$241k
- Technical Lead Manager - Training Runtime, Data(set) MovementOpenAIOn-site · Lead / Manager · Infrastructure$295k–$445k