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 exceptional ML engineers who can turn RL research ideas into working training systems, evals, environment and rewards. You will work across research and engineering to make post-training methods reliable, measurable, and fast to iterate on.
Responsibilities
- Build and improve RL training pipelines for language model based agents.
- Translate research ideas into working implementations, including reward functions, verifiers, environment interfaces, rollout pipelines, and evaluation harnesses.
- Design experiments that test whether RL methods are actually improving model behavior, sample efficiency, robustness, or generalization.
- Create quality monitoring tools for RL experiments, including regression tests, eval suites, and reward-hacking checks.
- Debug unstable training runs, diagnose poor learning dynamics, and identify whether failures come from algorithms, rewards, data, infrastructure, or evals.
- Build 0→1 systems for new RL workflows, then harden them into reusable infrastructure.
- Improve the reliability, reproducibility, and speed of experimentation across RL projects.
- Own technically ambiguous projects end to end, from problem framing through implementation, evaluation, and iteration.
Minimum Requirements
- Strong practical ML engineering ability, demonstrated through shipped systems, open-source projects, competitions, independent projects, or equivalent experience.
- Hands-on experience building, training, evaluating, or debugging ML systems.
- Strong programming ability in Python and experience with at least one major ML framework such as PyTorch or JAX.
- Working understanding of reinforcement learning, supervised learning, optimization, and modern deep learning.
- Ability to independently take an ambiguous technical problem and drive it to a working implementation..
- Ability to collaborate closely with researchers while maintaining high engineering standards.
- Experience building systems that are reliable, maintainable, and usable by other technical team members.
- Clear written and verbal communication.
Nice to have
- Experience supporting research teams or fast-moving ML teams.
- Expertise in building experiment tracking, evaluation platforms, dataset/versioning systems, or reproducibility infrastructure.
- Experience at a high engineering bar organization where reliability, ownership, and code quality were central.
- Experience reducing operational complexity in systems that had become brittle, slow, or hard to debug.
Role specific location policy
- This role is based in our San Francisco office; For exceptional candidates we are willing to consider a hybrid arrangement
Compensation
The expected salary range for this position is $300,000 - $500,000 USD