Member of Technical Staff - RL Infrastructure

Vmax is an applied research lab developing AI capable of open-ended learning. The Member of Technical Staff will build and operate the systems layer for reinforcement learning at scale, including distributed rollouts, training orchestration, inference, evaluations, data pipelines, observability, and reliability. The role enables researchers and applied ML engineers to run, debug, and reproduce large-scale RL experiments.

Responsibilities:

• Build infrastructure for distributed RL training and inference across thousands of GPUs
• Improve the reliability, debuggability, and throughput of RL experiments
• Build interfaces that allow researchers and applied ML engineers to launch, inspect, compare, and reproduce experiments easily
• Own infrastructure projects end to end, from architecture and implementation through deployment, documentation, and long-term maintenance
• Identify and eliminate bottlenecks in training, rollout generation, eval execution, data movement, and cluster utilization
• Maintain engineering standards for RL infrastructure, including testing, observability, versioning, and reproducibility