New Grad AI/ML Companies UniversitiesFor Universities Sign in Sign upGet started free

Staff Machine Learning Research Engineer, Agent Post-training - Enterprise GenAI at Scale AI · LumiJobs

Back to jobs

Overview Company

Apply on Employer SiteApply Now

Scale AI· Posted 2026-09-01

Staff Machine Learning Research Engineer, Agent Post-training - Enterprise GenAI

San Francisco, CAfull timeOn Site

Apply on Employer Site

senior

$265k–$331k

H1B: yes

Scale AI Staff Machine Learning Research Engineer focused on building and optimizing training and inference frameworks for next-gen Agent RL training platforms. Responsible for post-training state-of-the-art models and collaborating with ML teams to accelerate research and development.

Responsibilities

Build, profile and optimize our training and inference framework.
Post-train state of the art models, developed both internally and from the community, to define stable post-training recipes for our enterprise engagements.
Collaborate with ML teams to accelerate their research and development, and enable them to develop the next generation of models and data curation..
Create a next-gen agent training algorithm for multi-agent/multi-tool rollouts.
Ideally you’d have:
At least 1-3 years of LLM training in a production environment
Passionate about system optimization
Experience with post-training methods like RLHF/RLVR and related algorithms like PPO/GRPO etc.
Ability to demonstrate know-how on how to operate the architecture of the modern GPU cluster
Experience with multi-node LLM training and inference
Strong software engineering skills, proficient in frameworks and tools such as CUDA, Pytorch, transformers, flash attention, etc.
Strong written and verbal communication skills to operate in a cross functional team environment.

Qualification

Strong software engineering skills

Required

1-3 years of LLM training in a production environment
Experience with post-training methods like RLHF/RLVR and related algorithms like PPO/GRPO
Ability to operate modern GPU cluster architecture
Experience with multi-node LLM training and inference
Strong software engineering skills
PhD or Masters in Computer Science or a related field

Benefits

Dental
Vision
PTO
Equity/RSU

Company

Scale AI

View company page

Source: greenhouse

Active within the past hour

More roles at Scale AI

Recruiting Coordinator, Contract

New York, NY

Product Security Engineer, Public Sector

San Francisco, CA

Executive Assistant

Washington, DC

Technical Program Manager, Public Sector

Washington, DC

Boost Your Interview Chances

Improve Resume Match Score

Free

Your Score

9.0

Strong Applicant

Must-Have Skills for This Role

Strong software engineering skills

Optimize my Resume

Apply Faster with Autofill

Free

Save your profile once. LumiJobs pre-fills the employer's application form so you stop re-typing the same answers on every site.

Apply With Autofill

Or skip LumiJobs's tools and submit directly on the employer's career site.

Apply on Employer Site