Staff ML Engineer: Agent Training & Environments

EngineersOfAI; San Francisco (CA); On-site; USD 170,000 - 260,000; Full time

Job summary

Labelbox is shaping the data infrastructure for frontier AI models, building environments, evaluators, and pipelines that enable scalable experimentation and rapid iteration. This role blends platform engineering with agent-focused post-training expertise, requiring engineers who can ship reliable systems at scale and drive architectural direction.

You will work on RL environments, verifiers, and fine-tuning pipelines, contributing to the throughput and quality of AI training workflows in a

Qualifications

• 3+ years shipping reliable systems used by customers.
• Strong system and API design judgment; able to make hard architecture calls.
• Experience shipping production code with coding agents daily.

Responsibilities

• Develop RL environments and tool surfaces for agentic tasks.
• Build verifiers, graders, and evaluation pipelines to measure progress.
• Create scalable training and serving infrastructures for research workloads.
• Enable large-scale experimentation with reliable fault tolerance and cost accounting.
• Collaborate with teams to drive architecture decisions and roadmap.

Skills

System design

API design

High throughput

Production code

Platform engineering

Job description

Labelbox is shaping the data infrastructure for frontier AI models, building environments, evaluators, and pipelines that enable scalable experimentation and rapid iteration. This role blends platform engineering with agent-focused post-training expertise, requiring engineers who can ship reliable systems at scale and drive architectural direction.

You will work on RL environments, verifiers, and fine-tuning pipelines, contributing to the throughput and quality of AI training workflows in a