Luma AI logo

Research Scientist / Engineer - Training Infrastructure

Luma AI

Redwood City, CAJobNo compensation foundPosted 4w agoVerified open 5 days ago

Most applications go out cold — see where you stand first. No sign-up to start.

At a glance

Compensation
No compensation found
Location
Redwood City, CA
Work Authorization
Not specified

Job overview

Luma AI is hiring a Research Scientist / Engineer - Training Infrastructure. The role builds distributed systems for training Luma's large‑scale multimodal models across thousands of GPUs, enabling researchers to focus on innovation atop reliable, efficient, and scalable infrastructure.

Key focus areas include Design, implement, and optimize efficient distributed training systems for models across thousands of GPUs., Research and implement advanced parallelization such as FSDP, Tensor Parallel, Pipeline Parallel, and Expert Parallel., and Build monitoring, visualization, and debugging tools for large‑scale training runs..

Important skills include Internetworking, Storage Systems, Linux Administration, Scripting, Containerization, and Cloud Infrastructure. Preferred (not required): Linux Systems Administration and Orchestration.

Skills & qualifications

RequiredNice to have

Skills

InternetworkingStorage SystemsLinux AdministrationScriptingContainerizationCloud InfrastructurePyTorchCUDADistributed SystemsAdvanced ParallelismTraining StabilityFSDPTensor ParallelPipeline ParallelExpert ParallelMonitoringVisualizationDebuggingGPU ClustersNetworkingNCCLMPIDistributed System OptimizationLinux Systems AdministrationOrchestration

Qualifications

Experience Training Foundation Models at ScaleExtensive Distributed PyTorch TrainingExperience Managing Training Runs Across 100+ GPUs

Full 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.

You've read the whole posting — now see how you match it.