Member of Technical Staff - RL Algorithms
San FranciscoOn-site
AI Summary
Develops new reinforcement learning algorithms for post-training large language models, adapting pre-LLM and modern LLM approaches to improve sample efficiency and robustness.
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
RL has become the de-facto method of post-training LLMs. We are limited by the sample efficiency of the current policy gradient algorithms in use today, and are looking for a talented researcher to weave together pre-LLM and post-LLM approaches to learning from experience.
Responsibilities
- Develop new RL algorithms for post-training language models.
- Adapt ideas from pre-LLM reinforcement learning, such as model-based RL, temporal abstraction, and value-based learning, to modern LLM and agentic settings.
- Establish empirical baselines and evaluation protocols for measuring sample efficiency, robustness, generalization, and reward exploitation in LLM RL.
- Analyze failure modes of RL-trained models, including reward hacking, mode collapse, over-optimization, exploration failures, and distribution shift.
- Collaborate with researchers working on environments, evals, interpretability, reward modeling, and infrastructure to turn algorithmic ideas into reliable training systems.
- Own and develop a research agenda within Vmax, from identifying promising directions to executing experiments and communicating results.
Minimum Requirements
- PhD or equivalent experience in machine learning, reinforcement learning, or a closely related field.
- Track record of research excellence, as demonstrated by publications, open source work, deployed AI systems, or other substantial technical contributions.
- Deep understanding of modern machine learning, especially reinforcement learning, representation learning, and large language models.
- Strong familiarity with LLM post-training methods.
- Experience designing and running rigorous ML experiments, including ablations, baselines, evaluation design, and failure analysis.
- Experience with large-scale ML infrastructure, distributed training, experiment tracking, data pipelines, and debugging unstable training runs.
- Expertise with Python and at least one major ML framework such as PyTorch or JAX.
- Ability to work independently on open-ended research problems and turn ambiguous ideas into concrete experimental programs.
Nice to have
- Experience developing new RL algorithms or improving existing ones in domains such as robotics, games, simulated control, language models, or agents.
- Experience with LLM pre-training.
- Strong understanding of reward modeling, verifiers, process supervision, outcome supervision, or automated evaluation systems.
- Demonstrated software engineering ability
- Strong communication skills, especially the ability to explain algorithmic ideas, empirical results, and research implications to both technical and non-technical audiences
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
JAXLarge Language ModelsLLM Post-trainingModel-based RLPythonPyTorchReinforcement LearningReward ModelingTemporal AbstractionValue-based Learning
Explore related jobs
More jobs at Vmax
- Member of Technical Staff - RL InfrastructureSan Francisco
- Member of Technical Staff - Open EndednessSan Francisco
- Member of Technical Staff - Mechanistic InterpretabilitySan Francisco
- Member of Technical Staff - Applied RLSan Francisco
- Research Fellowship - Open EndednessSan Francisco
- Research Fellowship - Mechanistic InterpretabilitySan Francisco
Similar JAX jobs
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.