L

Luma AI

Multimodal AI models for video generation

Research Scientist / Engineer - Reinforcement Learning Infrastructure

AI ResearchConfirmed live in the last 24 hours
Full-timeNo salary listedRemote

About the Job

Requirements

  • Hands-on experience post-training large language models with reinforcement learning, including PPO/GRPO-family methods, RLHF, and RLVR, at meaningful scale.
  • Extensive distributed PyTorch training and parallelism experience, including FSDP and tensor, pipeline, and expert parallelism, for foundation models.
  • Experience building reinforcement learning environments, reward functions, verifiers, or evaluation harnesses for LLM agents, including sandboxed execution and multi-turn tool use.
  • Deep familiarity with RL post-training frameworks such as veRL, OpenRLHF, and TRL; Ray orchestration; and rollout inference engines such as vLLM and SGLang.
  • Strong understanding of GPU clusters, networking, and communication libraries such as NCCL and MPI under mixed training and inference workloads.

Responsibilities

  • Design, build, and scale distributed RL post-training systems, orchestrating trainer, rollout, environment, and reward workloads across thousands of GPUs.
  • Build high-throughput rollout generation by integrating inference engines such as vLLM and SGLang, weight synchronization, and asynchronous or off-policy schemes.
  • Design reproducible, scalable RL environments for agentic, multi-step tasks, including sandboxed code execution, tool use, computer use, and multimodal interaction, scaling to millions of episodes.
  • Build reward infrastructure including verifiable and programmatic rewards, reward-model serving, LLM-as-judge pipelines, and defenses against reward hacking.
  • Develop evaluation, monitoring, and debugging tooling to keep large RL runs stable.
  • Advance training efficiency and stability, and turn new post-training ideas into production runs with researchers.

Desired Qualifications

  • Run RL training across 100 or more GPUs, including asynchronous or disaggregated trainer/rollout architectures.
  • Experience with containerization and orchestration, including Kubernetes and Ray, for large environment fleets and sandboxed workloads.
  • Research contributions in RL for LLMs or open-source contributions to RL training frameworks.

About the Company

Luma AI

Multimodal AI models for video generation

Luma AI builds multimodal foundation models that generate, understand, and operate in the physical world. Its Ray video models and UNI image models power the Luma app and Dream Machine, a tool set for turning briefs into cinematic video, product visuals, and other visuals. The models run through the Luma API and the app, with Luma Agents translating briefs into finished visuals and enterprise clients receiving Dream Lab support and forward-deployed creatives. Luma stands apart by combining video, image, and world models with a production-oriented toolkit (app, API, and enterprise services) and a focus on physical AI for robotics, all aimed at helping creators and brands produce high-quality content quickly at scale while pushing multimodal and physical AI research.

Company size
201-500
Company stage
Series C
Total funding
$1.1B
Headquarters
San Francisco, CA
Founded
2021

Recently Posted Jobs

Explore more opportunities at Luma AI

See more jobs