New Grad AI/ML Companies Sign in Get started free

Machine Learning Systems Research Engineer, Agent Post-training - Enterprise GenAI at Scaleai · Lumi · Lumi

Back to jobs

Overview Company

Apply on Employer SiteApply Now

Scaleai· Posted 2026-05-26

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

San Francisco, CAfull timeOn Site

senior

H1B: yes

Scaleai The Machine Learning Systems Research Engineer at Scaleai builds and optimizes training and inference frameworks for next-generation agent RL training platforms. This role supports large-scale training and integrates state-of-the-art technologies to optimize ML systems for enterprise clients.

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

1-3 years LLM training in productionStrong software engineering skills

Required

1-3 years LLM training in production
Experience with post-training methods like RLHF/RLVR and 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 related field

Benefits

Dental
Vision
PTO
Equity/RSU

Company

Scaleai

View company page

Source: greenhouse

Active within the past hour

More roles at Scaleai

[2026] Applied Scientist - PhD Intern

San Mateo, CA

Boost Your Interview Chances

Improve Resume Match Score

Free

Your Score

8.6

Top Applicants

Must-Have Skills for This Role

1-3 years LLM training in productionStrong software engineering skills

Optimize my Resume

Apply Faster with Autofill

Free

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

Apply With Autofill

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

Apply on Employer Site