Machine Learning Research Engineer, Agents - Enterprise GenAI

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

Benefits offered by this job

Health insurance

Dental insurance

Vision insurance

Retirement benefits

Learning and development stipend

Generous PTO

Commuter stipend

Job summary

A leading AI technology company is seeking a Machine Learning Research Engineer to develop state-of-the-art models and algorithms for enterprise clients. The role involves training ML models, researching cutting-edge algorithms, and building high-performing agents using proprietary methods. Candidates should have experience with LLMs, relevant publications, and a degree in Computer Science. Competitive compensation and benefits offered.

Qualifications

• 1–3 years of building with LLMs in a production environment.
• Experience with related algorithms such as PPO/GRPO.
• Publications in top conferences within last two years.

Responsibilities

• Train state‑of‑the‑art models for enterprise customers.
• Research algorithms to integrate into training stack.
• Build agents to automatically hill‑climb datasets.

Skills

Building with LLMs

Experience with post-training methods like RLHF/RLVR

Research in cutting-edge algorithms

Education

Ph.D. or Master’s in Computer Science

Job description

Machine Learning Research Engineer, Agents - Enterprise GenAI

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 Agent ML RE, you will be applying our Agent RL Training + Building algorithms to real‑life enterprise datasets across our clients and benchmarks, creating best‑in‑class agents that achieve state‑of‑the‑art results through a combination of post‑training and agent‑building algorithms.

You will

• 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, defining highly performant tools, multi‑agent systems, and complex rewards.

Ideally you’d have

• 1–3 years of building with LLMs in a production environment.
• Experience with post‑training methods like RLHF/RLVR and related algorithms such as PPO/GRPO.
• Publications in top conferences such as NeurIPS, ICLR, or ICML within the last two years.
• Ph.D. or Master’s in Computer Science or a related field.

Compensation

The base salary range for this full‑time position in the locations of San Francisco, New York, Seattle is $180,600—$315,000 USD. The role may also include equity and benefits, including health, dental, vision, retirement benefits, a learning and development stipend, generous PTO, and a commuter stipend.

Equal Employment Opportunity Statement

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 providing reasonable accommodations to applicants with physical and mental disabilities.