Vmax AI Corp logo
Posted 5mo ago•SF

Member of Technical Staff - RL Algorithms

staffOn-site (San Francisco)Salary undisclosed
Required Skills
PythonPyTorchLLMs
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

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,

Similar Openings in Other

View all in category➔