Research Scientist / Engineer - Training Infrastructure

Luma

• $150K — $180K *
Information Technology
Less than 5 years of experience
Job Overview by Ladders

Qualifications

  • 5+ years experience in distributed systems and model training with PyTorch.
  • Strong knowledge of advanced parallelization techniques (FSDP, Tensor Parallel).
  • Proficient in GPU clusters, networking, and storage systems.
  • Experience with distributed-system optimization and communication libraries (NCCL, MPI).
  • Familiarity with Linux systems administration preferred.

Responsibilities

  • Design and optimize distributed training systems for thousands of GPUs.
  • Research and implement advanced parallelism techniques.
  • Develop monitoring, visualization, and debugging tools for training runs.
  • Enhance training stability, convergence, and resource utilization across clusters.
  • Conduct diagnostics on the existing training stack to identify stability issues.
  • Deliver measurable improvements in parallelization or stability during initial project phases.
  • Establish robust monitoring tools for large run efficiency and reliability.

Benefits

  • Flexible working arrangements to accommodate diverse lifestyles.
  • Opportunities for professional development in cutting-edge technologies.
  • Collaborative and innovative workplace culture focused on research and development.
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.

Similar Jobs

More Jobs at Luma

More Information Technology Jobs

Find similar Research Scientist / Engineer - Training Infrastructure jobs: