# Staff ML Engineer, Agent Training & Environments

**Company:** [Labelbox](https://hotfix.jobs/companies/labelbox)
**Location:** San Francisco, CA
**Role:** ML Engineering
**Salary:** $250k – $280k/yr
**Experience:** 7+ years
**Skills:** Python, rl, sft, ppo, dpo, grpo, Kubernetes, GCP, LLMs, Distributed Systems, ML Infrastructure
**Posted:** 2026-07-29

> Build RL environments, verifiers, fine-tuning pipelines, and eval systems for frontier AI agents at Labelbox. Requires deep RL post-training experience (SFT + RL methods), strong Python/systems engineering, and the ability to ship production infrastructure at high velocity.

## Job Description

## What you'll work on

- RL environments for agentic tasks: task definitions, tool surfaces, state and reset semantics, reward design — and the harness that runs thousands of them in parallel.
- Verifiers and graders: programmatic checks, LLM judges, rubric pipelines, pass@k scoring. Deciding what "the agent succeeded" means, and making that judgment trustworthy at scale.
- Fine-tuning pipelines that turn evaluation signals into measurable agent improvements — SFT and RL, from data collection through training to checkpoint evaluation.
- Eval systems that run millions of agent trajectories to measure model and product quality.
- Training and serving infrastructure that scales to the throughput frontier labs need: multi-launcher orchestration, long-running job fault tolerance, cost accounting.

## What we're looking for

### As an engineer
- A 3+ year track record of shipping systems that customers and other engineers still rely on.
- Exceptional throughput, without the quality tax. You ship a lot, you review a lot, and the v1 you ship becomes the foundation the rest of the team builds on.
- Strong system and API design judgment. Hard architecture calls land with you: you make them, defend them under pressure, and update fast when someone else is right.
- You ship production code with coding agents daily. You know where they break and what it takes to make them reliable, and you use that to move the whole team faster.
- You build the substrate other people's work runs on — tooling, CI, harnesses, libraries — and you treat that as the job, not a distraction from it.
- You move fast in ambiguous, startup-pace environments, with influence over authority.
- Deep proficiency in Python, and comfort across the rest of the stack.

### As an RL post-training practitioner
- You have fine-tuned models for agentic tasks and made them measurably better. SFT plus at least one RL method (GRPO, PPO, DPO, or similar) in production.
- You have built environments agents operate in, and you know why reward and task design is where most of the difficulty actually lives.
- You have designed verifiers or graders for open-ended work, and you know how they get gamed.
- You debug training runs forensically and methodically.
- You reason about compute-economics. You know what an experiment costs, when a run is not worth finishing, and how to get the same signal for a tenth of the spend.
- You write up what you learned so it changes what the team does next.

## Nice to have
- Experience with agent harnesses and coding agents as subjects of training and evaluation.
- Multi-tenancy and isolation for untrusted agent execution: sandboxing, egress control, credential handling.
- Background in production distributed systems, ML infrastructure, or data systems at scale.
- Experience working directly with frontier labs or other highly technical customers.

## Our Technology Stack
- Frontend: React.js with Redux, TypeScript
- Backend: Node.js, TypeScript, Python, some Java & Kotlin
- APIs: GraphQL
- Cloud & Infrastructure: Google Cloud Platform (GCP), Kubernetes
- Databases: MySQL, Spanner, PostgreSQL
- Queueing / Streaming: Kafka, PubSub

## Similar roles

- [Staff Software Engineer, AI Data Platform](https://hotfix.jobs/jobs/1914a72f-f09f-498a-8edf-51b951be967a) - Labelbox - San Francisco, CA - $250k – $280k/yr
- [Staff Software Engineer, Agentic Platform](https://hotfix.jobs/jobs/63ce2244-1c65-4c7d-820a-ca8afb78bb60) - Axion - San Francisco, CA - $250k – $270k/yr
- [Member of Research Staff, Optimization](https://hotfix.jobs/jobs/cd745934-2bf2-444e-b0b7-dc259cf569a4) - The Voleon Group - Berkeley, CA - $250k – $275k/yr
- [Member of Technical Staff — RL Research](https://hotfix.jobs/jobs/72e143f9-9ecc-422b-b67d-d6c97094fbaa) - Nuance Labs - Seattle, WA - $250k – $350k/yr
- [Member of Technical Staff — Model Optimization and Inference](https://hotfix.jobs/jobs/c81df2dc-c760-4cc8-9c23-2aa36efed90e) - Nuance Labs - Seattle, WA - $250k – $350k/yr

**Apply:** https://hotfix.jobs/jobs/faee3b40-f74e-43a6-97d5-b2eb7571cc64
**Canonical:** https://hotfix.jobs/jobs/faee3b40-f74e-43a6-97d5-b2eb7571cc64