Machine Learning Research Engineer (GenAI)

TL;DR

Machine Learning Research Engineer (GenAI): Developing and optimizing post-training algorithms for complex enterprise AI agents with an accent on RLHF and agent-building algorithms. Focus on creating best-in-class agents that achieve state-of-the-art results through the integration of proprietary research and large-scale enterprise datasets.

Location: Must be based in San Francisco, CA; New York, NY; or Seattle

Salary: $264,800 - $331,000 USD

Company

Scale AI is a leading AI data foundry that accelerates the development of generative AI, defense applications, and autonomous vehicles.

What you will do

• Train state-of-the-art models developed internally and from the community for enterprise customers.
• Research cutting-edge algorithms to integrate directly into the training stack.
• Build agents using proprietary algorithms to automatically hill climb datasets.
• Define highly performant tools, multi-agent systems, and complex reward functions.

Requirements

• 1-3 years of experience building with LLMs in a production environment.
• Experience with post-training methods like RLHF/RLVR and algorithms such as PPO/GRPO.
• PhD or Masters in Computer Science or a related field.
• Recent publications in top conferences such as NEURIPS, ICLR, or ICML.
• Must be located in San Francisco, New York, or Seattle

Culture & Benefits

• Comprehensive health, dental, and vision coverage.
• Equity-based compensation.
• Retirement benefits and a learning and development stipend.
• Generous PTO and commuter stipends for eligible roles.