Remote Jobs RockRemote Jobs Rock

Research Scientist / Engineer – Reinforcement Learning Infrastructure

🕒 12 days ago
Reinforcement LearningLLM Post-trainingPPOGRPO
Apply Now

📜 Description

  • Design and scale distributed RL post-training systems across trainer, rollout, environment, and reward workloads.
  • Build high-throughput rollout generation using inference engines such as vLLM and SGLang, with weight synchronization and asynchronous or off-policy methods.
  • Create reproducible agent environments for sandboxed code execution, tool use, computer use, and multimodal interaction.
  • Develop reward infrastructure, including programmatic rewards, reward-model serving, LLM-as-judge pipelines, and defenses against reward hacking.
  • Build evaluation, monitoring, and debugging tools to keep large RL runs stable.
  • Improve training efficiency and stability and bring post-training research ideas into production runs.

🛠️ Requirements

  • Hands-on experience post-training LLMs with RL (PPO/GRPO-family, RLHF, RLVR) at meaningful scale.
  • Extensive distributed PyTorch training and parallelism (FSDP, Tensor/Pipeline/Expert Parallel) for foundation models.
  • Experience building RL 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 (veRL, OpenRLHF, TRL, Ray orchestration) and rollout inference engines (vLLM, SGLang).
  • Strong understanding of GPU clusters, networking, and communication libraries (NCCL, MPI) under mixed training and inference workloads.
  • Running RL training across 100+ GPUs, including asynchronous or disaggregated trainer/rollout architectures.
  • Containerization and orchestration (Kubernetes, Ray) for large environment fleets and sandboxed workloads.
  • Research contributions in RL for LLMs, or open-source contributions to RL training frameworks.
Full job description

You'll build the systems that make reinforcement learning work at frontier scale — coupling policy optimization with large fleets of inference workers, agentic environments, and the reward and verification systems that turn model behavior into learning signal. RL is how Luma's models go from capable to useful.

RL at scale is a full-loop systems problem: training, rollout generation, environment execution, and reward computation running concurrently across thousands of GPUs, all needing to stay fast, stable, and correct together. It fits someone who has lived this — post-trained LLMs with RL, built environments and verifiers, and debugged asynchronous rollout pipelines at scale. If you haven't operated RL at real scale, this will be deep water.

What You'll Own

  • 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, integrating inference engines (vLLM, SGLang), weight synchronization, and asynchronous/off-policy schemes.

  • Design RL environments for agentic, multi-step tasks — sandboxed code execution, tool use, computer use, multimodal interaction — reproducible and scalable to millions of episodes.

  • Build reward infrastructure: verifiable/programmatic rewards, reward-model serving, LLM-as-judge pipelines, and defenses against reward hacking.

  • Develop the evaluation, monitoring, and debugging tooling that keeps large RL runs stable.

  • Advance training efficiency and stability, and turn new post-training ideas into production runs with researchers.

First 90 Days

One way the first 90 could unfold.

  • Days 1–30 — Immerse & Diagnose: Learn the current RL stack and where throughput, stability, or correctness break.

  • Days 30–60 — Ship & Validate: Improve a piece of the loop (rollout throughput, reward infra, or an environment) and prove it on a real run.

  • Days 60–90 — Scale & Systemize: Harden the full loop across thousands of GPUs and asynchronous architectures.

What You Bring

  • Hands-on experience post-training LLMs with RL (PPO/GRPO-family, RLHF, RLVR) at meaningful scale.

  • Extensive distributed PyTorch training and parallelism (FSDP, Tensor/Pipeline/Expert Parallel) for foundation models.

  • Experience building RL 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 (veRL, OpenRLHF, TRL, Ray orchestration) and rollout inference engines (vLLM, SGLang).

  • Strong understanding of GPU clusters, networking, and communication libraries (NCCL, MPI) under mixed training and inference workloads.

Nice to Have

  • Running RL training across 100+ GPUs, including asynchronous or disaggregated trainer/rollout architectures.

  • Containerization and orchestration (Kubernetes, Ray) for large environment fleets and sandboxed workloads.

  • Research contributions in RL for LLMs, or open-source contributions to RL training frameworks.

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.

Wayve

Senior Machine Learning Engineer, Driving Product

🕒 4 days ago
Wayve👥 501 - 1000 employees🏢 Computer Software

Develop and ship autonomous-driving ML models for L2–L4 vehicles, owning data curation, training, evaluation, and on-road iteration to improve driving behaviours and real-world performance.

PythonDeep LearningMachine Learning FrameworksAutonomous Driving
Taboola

Senior Machine Learning Engineer

🕒 3 days ago
Taboola👥 1001 - 5000 employees🏢 Computer Software

Build and operate recommendation, ranking, and personalization models that shape advertiser campaign performance, while owning production ML systems and the evaluation frameworks used to test and ship them safely.

PythonJavaPytorchTensorflow
Doctolib

Senior IA Inference Engineer (x/f/m)

🕒 4 days ago
Doctolib👥 1001 - 5000 employees🏢 Computer Software

Own and scale the self-hosted LLM inference platform powering healthcare AI products, optimizing model serving performance, reliability, security, and cost across GPU infrastructure.

PythonTerraformLLM Inference ServingVllm
Wayve

Senior Machine Learning Engineer, ADAS

🕒 4 days ago
Wayve👥 501 - 1000 employees🏢 Computer Software

Train and ship computer vision and 3D perception models for ADAS, owning work across data, evaluation, and iteration to improve real-world driving performance under production constraints.

Machine LearningApplied ML EngineeringComputer VisionDeep Learning

Trusted by Remote Workers