Quick overview
Job description
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
Similar jobs
- AL

Staff Software Engineer – Distributed SystemsRecently posted Auto Apply
Alluxio · Foster City · On-site 🇺🇸
TechnologySalary not listed - AL

Software Engineer – Distributed Systems Auto Apply
Alluxio · Foster City · On-site 🇺🇸
C++JavaSystem DesignTechnologySalary not listed - AL

Senior Software Engineer - Distributed System Development Auto Apply
Alluxio · Foster City · On-site 🇺🇸
TechnologySalary not listed - AL

Staff Software Engineer - Distributed System Development Auto Apply
Alluxio · Foster City · On-site 🇺🇸
TechnologySalary not listed - AF
Senior Systems Engineer Auto Apply
Afero · Los Altos · Hybrid 🇺🇸
Fleet ManagementIoTAndroid+1Technology$150k - $220k/yr - 9D

Senior Embedded Software Engineer Auto Apply
908 Devices · Burlington · Hybrid 🇺🇸
Microsoft ExcelC++Python+3TechnologySalary not listed - SE
Senior Staff Software Engineer – Compute EngineeringRecently posted Auto Apply
ServiceNow · Santa Clara, CALIFORNIA · Hybrid 🇺🇸
PythonJavaMicroservices+8TechnologySalary not listed - SE
Senior Software Engineer – Compute EngineeringRecently posted Auto Apply
ServiceNow · Santa Clara, CALIFORNIA · Hybrid 🇺🇸
PythonJavaAWS+7TechnologySalary not listed - VE
Junior Manufacturing Software EngineerRecently posted Auto Apply
Veolia Environnement SA · Boulder, CO · On-site 🇺🇸
JiraTechnologySalary not listed - ST

Senior Embedded Software EngineerRecently posted Auto Apply
Sapsol Technologies Inc · Clearwater · On-site 🇺🇸
SVNGitEmbedded Systems+3TechnologySalary not listed - AF

Application Engineer (remote U.S.-based ONLY) Auto Apply
Absolute Facility Solutions, LLC · Lago Vista, TX · Remote 🇺🇸
SQLETLJava+4Technology$90k - $120k/yr - EC
Senior Vice President, Infrastructure Engineering (Mainframe Communications Systems Software EngineerRecently posted Auto Apply
ewminds consulting Services · Jersey City, NJ · On-site 🇺🇸
ITILTalent ManagementTechnologySalary not listed
