Member of Technical Staff - Applied RL
San FranciscoOn-site
AI Summary
A hands-on ML engineer role that designs, builds, and scales reliable RL training and evaluation systems for language model agents, turning research ideas into production-ready pipelines.
About this role
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
Skills
Dataset/versioning SystemsDeep LearningEvaluation PlatformsExperiment TrackingJAXOptimizationPythonPyTorchReinforcement LearningSupervised Learning
Explore related jobs
More jobs at Vmax
Jobs in San Francisco
Strategic Sourcing Manager, Engineering Technology & SoftwareHarvey · San Francisco- Senior Software Engineer, CardsChime Financial, Inc · San Francisco, CA
VC Partnerships LeadCAMPFIRE · San Francisco
Executive Sous ChefAltius Search Group · San Francisco, California
Senior Data Scientist, Fleet HealthGridware · San Francisco, California
AI Factory Partnerships Lead - OEM & Infrastructure AlliancesFireworks · San Francisco
Browse these categories
Market data for this role
All reports →- SeriesRole reportsOne role family at a time: how many openings, what changed this week, who is hiring, what it pays.
- SeriesSalary reportsWhat employers publish in job postings, by level and workplace. Not self-reported pay.
- Market overviewState of tech hiring, September 2026: up 4.8%Tech hiring rose 4.8% month over month in September 2026, with 411,122 new listings. Customer support and account executive roles led the growth.