Member of Technical Staff - RL Infrastructure
About Vmax
Vmax is an applied research lab developing AI capable of open-ended learning. We are building systems to exceed humans in all capacities by optimising beyond the local maxima of learning from human expertise.
About the role
This role is for strong infrastructure engineers who can build the systems layer for RL at scale: distributed rollouts, training orchestration, inference, evals, data pipelines, observability, and reliability. You will create the durable platform that 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 clu
