Staff ML Engineer: Agent Training & Environments

WeHireYou

San Francisco (CA)

On-site

USD 180,000 - 280,000

Full time

14 days+
Application generator

Don’t send a generic resume — generate a resume and cover letter tailored to this exact role.

Get past ATS filters

Job summary

Labelbox is the RL data factory for advancing frontier agent capabilities. We build the data, environments, and evaluations that frontier labs use to train and judge their agents.

This Staff ML Engineer role sits where training meets infrastructure, running experiments and building environments agents act in, verifiers that decide success, and pipelines that turn signals into a better model.

Qualifications

  • 3+ years shipping systems used by customers and engineers.
  • Strong system and API design judgment for hard architectural calls.
  • Experience shipping production code and maintaining reliability in complex platforms.

Responsibilities

  • RL environments for agentic tasks and tooling.
  • Verifiers and graders to judge agent success at scale.
  • Fine-tuning pipelines from data collection to evaluation.
  • Eval systems handling millions of agent trajectories.
  • Training and serving infra to support frontier labs.

Skills

System design
API design
Production code
High throughput

Job description

Labelbox is the RL data factory for advancing frontier agent capabilities. We build the data, environments, and evaluations that frontier labs use to train and judge their agents.

This Staff ML Engineer role sits where training meets infrastructure, running experiments and building environments agents act in, verifiers that decide success, and pipelines that turn signals into a better model.

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Staff ML Engineer: Agent Training & Environments
Staff ML Engineer: Agent Training & Environments

EngineersOfAI • San Francisco (CA)

On-site
USD 170,000 - 260,000
Staff ML Engineer: Frontier Agent Training & Environments
Staff ML Engineer: Frontier Agent Training & Environments

B Capital • San Francisco (CA)

Hybrid
USD 250,000 - 280,000
Hybrid work model (3 days in office)
Career advancement opportunities
Staff AI Infrastructure Engineer (RL Environments)
Staff AI Infrastructure Engineer (RL Environments)

AI Chopping Block, Inc. • San Francisco (CA), Northern (KY)

Hybrid
USD 140,000 - 200,000
Staff ML Engineer, Agent Training & Environments
Staff ML Engineer, Agent Training & Environments

EngineersOfAI • San Francisco (CA)

On-site
USD 170,000 - 260,000
Staff Software Engineer — RL Environments & Tools
Staff Software Engineer — RL Environments & Tools

Labelbox • San Francisco (CA)

On-site
USD 140,000 - 200,000
Staff ML Engineer, Agent Training & Environments
Staff ML Engineer, Agent Training & Environments

WeHireYou • San Francisco (CA)

On-site
USD 180,000 - 280,000
Staff Engineer, Reinforcement Learning Environments
Staff Engineer, Reinforcement Learning Environments

Labelbox • San Francisco (CA), Northern (KY)

Hybrid
USD 140,000 - 200,000
Staff ML Engineer, Agent Training & Environments
Staff ML Engineer, Agent Training & Environments

B Capital • San Francisco (CA)

On-site
USD 250,000 - 280,000
Hybrid work model (3 days in office)
Career advancement opportunities
Member of Technical Staff
Member of Technical Staff

Labelbox • San Francisco (CA)

On-site
USD 140,000 - 200,000
Member of Technical Staff
Member of Technical Staff

AI Chopping Block, Inc. • San Francisco (CA), Northern (KY)

Hybrid
USD 140,000 - 200,000