Machine Learning Research Engineer, Agent Data Foundation - Enterprise GenAI

Scale AI; United States; Hybrid; USD 180,600 - 315,000; Full time

Benefits offered by this job

Equity

Comprehensive benefits package

Learning and development stipend

Generous PTO

Commuter stipend

Job summary

A leading AI data foundry in the United States is seeking a Machine Learning Research Engineer to work on advanced AI applications. You will build synthetic data pipelines, create actionable insights, and train state-of-the-art models for enterprise clients. Candidates should have 3+ years of experience with LLMs and a relevant advanced degree. The position offers a competitive salary, equity, and comprehensive benefits.

Qualifications

• 3+ years of building with LLMs in a production environment.
• Experience with constructing high-quality data to improve LLM/Agent.
• Publications in top conferences like NEURIPS, ICLR, or ICML in the last two years.

Responsibilities

• Build synthetic data pipelines for RL post-training.
• Create agents for actionable insights from production traces.
• Contribute to agent-building product with proprietary algorithms.
• Train state-of-the-art models for enterprise customers.

Skills

LLMs experience

Data pipeline construction

Agent building

Synthetic data generation

Education

PhD or Masters in Computer Science or related field

Job description

Machine Learning Research Engineer, Agent Data Foundation - Enterprise GenAI

We are looking for a Machine Learning Research Engineer, Agent Data Foundation - Enterprise GenAI at Scale AI. AI is becoming vitally important in every function of our society. At Scale, our mission is to accelerate the development of AI applications. For 9 years, Scale has been the leading AI data foundry, helping fuel the most exciting advancements in AI, including generative AI, defense applications, and autonomous vehicles. With our recent investment from Meta, we are doubling down on building out state‑of‑the‑art post‑training algorithms to reach the performance necessary for complex agents in enterprises around the world.

The Enterprise ML Research Lab works on the front lines of this AI revolution. We are working on an arsenal of proprietary research, tools, and resources that serve all of our enterprise clients. As an MLRE on the Data Foundation team, you’ll work on cutting‑edge research to define the data flywheel that makes the whole machine move. This includes research around synthetic environments from task definitions, building agents for trace analysis, and contributing to a cutting‑edge framework that automatically hill‑climbs agent‑building from an eval set. This will involve creating best‑in‑class agents that achieve state‑of‑the‑art results through a combination of post‑training + agent‑building algorithms.

Responsibilities

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

Preferred Qualifications

• 3+ years of building with LLMs in a production environment
• Clear experience with constructing high‑quality data to use to improve an LLM/Agent
• 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

Compensation

Base salary range: $180,600 – $315,000 USD (location based). Eligible roles also receive equity and a comprehensive benefits package that includes health, dental and vision coverage, retirement benefits, a learning and development stipend, and generous PTO. This role may also be eligible for additional benefits such as a commuter stipend.

Equal Employment Opportunity

We believe that everyone should be able to bring their whole selves to work, which is why we are proud to be an inclusive and equal opportunity workplace. We are committed to equal employment opportunity regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability status, gender identity or Veteran status. We are committed to working with and providing reasonable accommodations to applicants with physical and mental disabilities. If you need assistance and/or a reasonable accommodation in the application or recruiting process due to a disability, please contact us at [e-mail].

This position will be located in the San Francisco, New York, or Seattle market based on the location posting. The role is open, and we encourage early applications. Referrals increase your chances of interviewing at Scale AI by 2x.