Member of Technical Staff - RL Infrastructure

Vmax is an applied research lab developing AI capable of open-ended learning. The role focuses on building and maintaining the infrastructure for reinforcement learning at scale, enabling researchers and applied ML engineers to conduct large-scale 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