Machine Learning Research Engineer, Agents - Enterprise GenAI
$265k - $331k • San Francisco, CA; New York, NY
Posted 2mo ago
About the job
Scale is seeking a Machine Learning Research Engineer focused on Agents for its Enterprise GenAI team. This role will be instrumental in accelerating the development of AI applications by working on state-of-the-art post-training algorithms for complex enterprise agents. You will apply proprietary Agent RL Training + Building algorithms to real-world enterprise datasets and benchmarks, aiming to create best-in-class Agents that achieve state-of-the-art results. If you are passionate about shaping the future of GenAI, this is an opportunity to contribute to cutting-edge research and development.
Responsibilities
- Train state-of-the-art models for deployment to enterprise customers.
- Research and integrate cutting-edge algorithms into the training stack.
- Build agents using proprietary algorithms to optimize datasets, including defining tools, multi-agent systems, and rewards.
Requirements
- 1-3 years of experience building with LLMs in a production environment.
- Experience with post-training methods like RLHF/RLVR and algorithms like PPO/GRPO.
- Publications in top conferences (e.g., NEURIPS, ICLR, ICML) within the last two years.
- PhD or Master's degree in Computer Science or a related field.
Benefits
- Base salary
- Equity compensation
- Comprehensive health, dental, and vision coverage
- Retirement benefits
- Learning and development stipend
- Generous PTO
- Commuter stipend