Machine Learning Research Engineer, Agent Data Foundation - Enterprise GenAI

About the role

At Scale, our mission is to accelerate the development of AI applications. The Enterprise ML Research Lab works on the front lines of this AI revolution, developing proprietary research, tools, and resources for our enterprise clients. As an ML Research Engineer on the Data Foundation team, you’ll work on cutting-edge research to define the data flywheel that powers our AI systems. This includes research around synthetic environments from task definitions, building agents for trace analysis, and contributing to a framework that automatically optimizes agent-building from an eval set. You will create best-in-class agents that achieve state-of-the-art results through post-training and agent-building algorithms, shaping the future of the modern GenAI movement.

Responsibilities

• Build synthetic data pipelines to generate enterprise environments for RL post-training.
• Create agents to convert production traces into actionable insights for improving agents.
• Contribute to our agent-building product, which constructs other agents using coding agents and proprietary algorithms.
• Train state-of-the-art models, both internally developed and from the community, to deploy to enterprise customers.

Requirements

• 3+ years of experience building with LLMs in a production environment.
• Experience constructing high-quality data to improve LLMs or agents.
• Publications in top conferences such as NeurIPS, ICLR, or ICML within the last two years.
• PhD or Master’s in Computer Science or a related field.

Benefits

• Comprehensive health, dental, and vision coverage.
• Retirement benefits.
• Learning and development stipend.
• Generous PTO.
• Eligibility for additional benefits such as a commuter stipend.

Pay

The base salary range for this full-time position in San Francisco, New York, or Seattle is $264,800—$331,000 USD. Compensation packages at Scale for eligible roles include base salary, equity, and benefits. The final salary will be determined during the interview process based on work location, job-related skills, experience, qualifications, interview performance, and relevant education or training.

About Us

At Scale, our mission is to develop reliable AI systems for the world's most important decisions. We provide high-quality data and full-stack technologies that power leading AI models and help enterprises and governments build, deploy, and oversee impactful AI applications. Our clients include industry leaders like Meta, Ernst & Young, Mayo Clinic, Time Inc., the Government of Qatar, and U.S. government agencies, including the Army and Air Force.