Note: The job is a remote job and is open to candidates in USA. Scale AI is a leading AI data foundry focused on accelerating the development of AI applications. The Machine Learning Research Engineer will work on applying agent reinforcement learning algorithms to enterprise datasets, creating best-in-class agents that achieve state-of-the-art results.

Responsibilities

• Train state of the art models, developed both internally and from the community, to deploy to our enterprise customers
• Research cutting edge algorithms to integrate directly into our training stack
• Build agents that leverage our proprietary agent-building algorithms to automatically hill climb datasets – including defining highly performant tools, multi-agent systems, and complex rewards

Skills

• 1-3 years of building with LLMs in a production environment
• Experience with post-training methods like RLHF/RLVR and related algorithms like PPO/GRPO etc
• Publications in top conferences such as NEURIPS, ICLR, or ICML within the last two years
• PhD or Masters in Computer Science or a related field

Benefits

• Equity based compensation, subject to Board of Director approval
• Comprehensive health, dental and vision coverage
• Retirement benefits
• A learning and development stipend
• Generous PTO
• A commuter stipend

Company Overview

Company H1B Sponsorship