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.
117open roles right now
- Senior Software Engineer - Runtime Platform, Robot SoftwareWayveHybrid · Senior · Infrastructure$210k–$267k
- Engineering Team Lead, StorageMistral AIOn-site · Lead / Manager · Infrastructure
- Member of Technical StaffLabelboxOn-site · Mid-level · Infrastructure$140k–$200k
- 3P Systems ArchitectOpenAIHybrid · Senior · Infrastructure$342k–$555k
- AI Infrastructure Engineer, pAGIOpenAIOn-site · Mid-level · Infrastructure$266k–$500k
- Senior Staff Engineer, Test Director (R5446)Shield AIOn-site · Staff+ · Infrastructure$172k–$258k
- Staff Software Engineer, Lakeflow Pipelines DRDatabricksOn-site · Staff+ · Infrastructure$192k–$260k
- Senior Engineer, Modelling and Simulation (R5702)Shield AIOn-site · Senior · Infrastructure
- Staff Software Engineer, RL EnvironmentsScale AIOn-site · Staff+ · Infrastructure$252k–$315k
- ML Infra Engineer, ModelingPhysical IntelligenceOn-site · Mid-level · Infrastructure
- Senior Principal Runtime EngineerSambaNova SystemsOn-site · Senior · Infrastructure
- Software Engineer, PretrainingCursor (Anysphere)On-site · Senior · Infrastructure
- Director of ASIC Design VerificationSambaNova SystemsOn-site · Lead / Manager · Infrastructure$270k–$340k
- Software Engineer, Compute InfrastructureOpenAIRemote (US, UK) · Senior · Infrastructure$230k–$405k
- Helix AI Engineer, Training PerformanceFigure AIOn-site · Senior · Infrastructure$200k–$400k
- AI Performance EngineerApplied IntuitionOn-site · Senior · Infrastructure$215k–$285k
- Software Engineer, ML Infrastructure PlatformNuroOn-site · Mid-level · Infrastructure$160k–$241k
- Senior Software Engineer, ML Infrastructure PlatformNuroOn-site · Senior · Infrastructure$194k–$291k
- Staff Software Engineer, HPCZooxOn-site · Staff+ · Infrastructure$201k–$315k
- 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 Intern - Fall 2026/Winter 2027SkydioOn-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$210k–$320k
- 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
- 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$177k–$270k
- Software Engineer - Platform Infrastructure (Rust, C++)xAIOn-site · Mid-level · Infrastructure$180k–$440k
- Staff Software Engineer - Infrastructure StorageLambdaHybrid · Staff+ · Infrastructure$314k–$465k
- Member of Technical Staff - RL Training FrameworkxAIOn-site · Mid-level · Infrastructure$180k–$440k
- 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
- Sr. Engineering Manager, AI RuntimeDatabricksOn-site · Lead / Manager · Infrastructure$229k–$297k
- Senior Software Engineer II (IC4) – AI Workload OrchestrationCoreWeaveOn-site · Senior · Infrastructure$207k–$275k