Quick Overview
Job Description
You'll build the distributed systems that train Luma's large-scale multimodal models across thousands of GPUs, so researchers can focus on innovation on top of reliable, efficient, scalable infrastructure.
This is hard PyTorch, CUDA, and distributed-systems work — advanced parallelism, training stability, and utilization across massive clusters. It fits an engineer who's solved real problems training foundation models at scale. If you haven't worked at the level of FSDP and multi-node training, this is the wrong depth.
What You'll Own
Design, implement, and optimize efficient distributed training systems for models across thousands of GPUs.
Research and implement advanced parallelization (FSDP, Tensor Parallel, Pipeline Parallel, Expert Parallel).
Build monitoring, visualization, and debugging tools for large-scale training runs.
Optimize training stability, convergence, and resource utilization across massive clusters.
First 90 Days
One way the first 90 could unfold.
Days 1–30 — Immerse & Diagnose: Learn the current training stack and where stability and utilization hurt at scale.
Days 30–60 — Ship & Validate: Land a parallelization or stability improvement that measurably helps a real training run.
Days 60–90 — Scale & Systemize: Build the monitoring and tooling that keeps large runs reliable and efficient.
What You Bring
Extensive distributed PyTorch training and parallelisms in foundation-model training.
Deep understanding of GPU clusters, networking, and storage systems.
Familiarity with communication libraries (NCCL, MPI) and distributed-system optimization.
Nice to Have
Strong Linux systems administration and scripting.
Experience managing training runs across 100+ GPUs.
Experience with containerization, orchestration, and cloud infrastructure.
About Luma: Luma's mission is to build unified general intelligence that can generate, understand, and operate in the physical world. We believe multimodality is critical for intelligence — the next step beyond language models comes from vision. Luma is an equal opportunity employer.
Similar jobs
- LU
Copy of Research Scientist / Engineer – Performance Optimization
NewAuto ApplyLumaai
London🇬🇧Hybrid4 hours agoCUDAPyTorchEngineering - GE
Clinical Bioinformatician (we have office locations in Cambridge, Leeds and London)
NewAuto ApplyGenomics England
London🇬🇧Hybrid5 hours agoBackground ChecksOnboardingPython - CR
Researcher, Online Threat Intelligence
NewAuto ApplyControl Risks
London🇬🇧Hybrid6 hours agoTechnology - EI
Scientist, Image Based Screening - Plant Biology Institute
NewAuto ApplyEllison Institute of Technology
Oxford🇬🇧On-site6 hours agoMachine LearningRobotics - EI
Scientist, Plant Biology and Functional Genomics - Plant Biology Institute
NewAuto ApplyEllison Institute of Technology
Oxford🇬🇧On-site6 hours agoRoboticsMarket Research - MQ
Quantum Sensing Systems Lead (UK)
NewAuto ApplyMesa Quantum Systems
London🇬🇧On-site16 hours agoRecruiting - HR
Research Data Expert
NewAuto ApplyHudson River Trading
London🇬🇧HybridYesterdayCornerstoneData PipelineProcurement+4 - FP
Senior Scientist, Bioanalytical Sciences
NewAuto ApplyFlagship Pioneering, Inc.
Cambridge🇬🇧HybridYesterdayRubyClinical ResearchCompliance+3 - DR
Fundamental Researcher - Compute Markets
NewAuto ApplyDRW
London🇬🇧HybridYesterdayExpressDerivativesFinancial Modeling+1 - EI
Scientist - Applications Team (Pathogen)
NewAuto ApplyEllison Institute of Technology
Oxford🇬🇧On-siteYesterdayOracleRoboticsInventory Management - EI
Senior Group Leader, Industry - Plant Biology Institute
NewAuto ApplyEllison Institute of Technology
Oxford🇬🇧On-siteYesterdayRoboticsComplianceIntellectual Property - WI
Lead Researcher, Verification Experience
NewAuto ApplyWise
London🇬🇧£102k - £133k/yrHybrid2 days ago