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.
95open roles right now
- Software Engineer, ML OpsCohereRemote (UK, EU) · Mid-level · Infrastructure
- Software Engineer, ML Workload PlatformCohereRemote (US, Canada) · Infrastructure$175k–$325k
- Research Engineer, InfrastructureHarveyOn-site · Senior · Infrastructure$230k–$450k
- Research Engineer, InfrastructureHarveyRemote, worldwide · Senior · Infrastructure$230k–$450k
- Research Engineer, InfrastructureHarveyOn-site · Mid-level · Infrastructure$230k–$450k
- Research Engineer, InfrastructureHarveyOn-site · Senior · Infrastructure$230k–$450k
- Member of Technical Staff (Software Engineer, Training Infrastructure)PerplexityOn-site · Mid-level · Infrastructure$220k–$405k
- Compiler Engineer - New GradSambaNova SystemsOn-site · Junior · Infrastructure$131k–$197k
- Staff Software Engineer - Infrastructure StorageLambdaHybrid · Staff+ · Infrastructure$314k–$465k
- Senior Staff Software Engineer, AI RuntimeDatabricksOn-site · Staff+ · Infrastructure$229k–$343k
- Software Engineer, CI/CD & ML InfrastructureOpenAIOn-site · Senior · Infrastructure$318k–$485k
- Staff Engineer, Modelling and Simulation (R6248)Shield AIOn-site · Staff+ · Infrastructure$150k–$230k
- Software Engineer, ML Training OptimizationZooxOn-site · Mid-level · Infrastructure$177k–$270k
- Senior/Staff Software Engineer, ML Training OptimizationZooxOn-site · Staff+ · Infrastructure$254k–$350k
- Research Scientist / Engineer – Reinforcement Learning InfrastructureLuma AIRemote (UK) · Senior · Infrastructure
- Research Scientist / Engineer – Performance OptimizationLuma AIRemote (EU) · Senior · Infrastructure
- Research Scientist / Engineer – Training InfrastructureLuma AIRemote (UK) · Senior · Infrastructure
- Research Scientist / Engineer – Training InfrastructureLuma AIOn-site · Senior · Infrastructure$195k–$395k
- Research Scientist / Engineer – Training InfrastructureLuma AIRemote (EU) · Senior · Infrastructure
- Senior Hardware Test and Reliability EngineerSkydioOn-site · Senior · Infrastructure$170k–$230k
- Research Engineer / Performance Engineer, RL Distributed SystemsAnthropicHybrid · Senior · Infrastructure$500k–$850k
- Member of Technical StaffLabelboxOn-site · Mid-level · Infrastructure$140k–$200k
- Senior Staff Engineer, Test Director (R5446)Shield AIOn-site · Staff+ · Infrastructure$172k–$258k
- Staff Software Engineer, Lakeflow Pipelines DRDatabricksOn-site · Staff+ · Infrastructure$192k–$260k
- 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
- Senior/Staff Software Engineer, Frontier ResearchDeepgramRemote (US) · Senior · Infrastructure$150k–$250k
- 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
- Senior/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
- Research Engineer, Code Agents InfraMistral AI · Senior · Infrastructure
- Industrial ComputeOpenAIRemote (US) · Infrastructure$177k–$285k
- Staff Software Engineer - StreamingDatabricksRemote (US) · Staff+ · Infrastructure$182k–$247k
- Member of Technical Staff, AI Training InfrastructureFireworks AIOn-site · Mid-level · Infrastructure$210k–$320k
- Staff Software Engineer, Environments InfrastructureAnthropicHybrid · Staff+ · Infrastructure$405k–$605k
- Test Validation EngineerNuroOn-site · Mid-level · Infrastructure$132k–$198k
- Kernel Engineer - New GradCerebrasRemote (US, Canada) · Junior · Infrastructure
- Machine Learning Infrastructure Engineer, Safeguards ResearchAnthropicHybrid · Mid-level · Infrastructure$350k–$500k
- Software Engineer, Compute Foundations SystemsOpenAIOn-site · Mid-level · Infrastructure$230k–$490k
- 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
- Senior Software Engineer II (IC4) – AI Workload OrchestrationCoreWeaveOn-site · Senior · Infrastructure$207k–$275k