Jobs Companies Labelbox Staff ML Engineer, Agent Training & Environments

Sobre este puesto de Staff ML Engineer, Agent Training & Environments en Labelbox

Labelbox · Híbrido · San Francisco Bay Area

Shape the Future of AI

At Labelbox, we're building the critical infrastructure that powers breakthrough AI models at leading research labs and enterprises. Since 2018, we've been pioneering data-centric approaches that are fundamental to AI development, and our work becomes even more essential as AI capabilities expand exponentially.

About Labelbox

We're the only company offering three integrated solutions for frontier AI development:

  1. Enterprise Platform & Tools: Advanced annotation tools, workflow automation, and quality control systems that enable teams to produce high-quality training data at scale
  2. Frontier Data Labeling Service: Specialized data labeling through Alignerr, leveraging subject matter experts for next-generation AI models
  3. Expert Marketplace: Connecting AI teams with highly skilled annotators and domain experts for flexible scaling

Why Join Us

  • High-Impact Environment: We operate like an early-stage startup, focusing on impact over process. You'll take on expanded responsibilities quickly, with career growth directly tied to your contributions.
  • Technical Excellence: Work at the cutting edge of AI development, collaborating with industry leaders and shaping the future of artificial intelligence.
  • Innovation at Speed: We celebrate those who take ownership, move fast, and deliver impact. Our environment rewards high agency and rapid execution.
  • Continuous Growth: Every role requires continuous learning and evolution. You'll be surrounded by curious minds solving complex problems at the frontier of AI.
  • Clear Ownership: You'll know exactly what you're responsible for and have the autonomy to execute. We empower people to drive results through clear ownership and metrics.

Role Overview

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 role sits where training meets infrastructure. You will run the experiments and build the systems that run them: environments agents act in, verifiers that decide whether they succeeded, and the fine-tuning pipelines that turn that signal into a better model. We're looking for someone who does both halves — the engineering throughput of a strong platform engineer, and real depth in post-training agents.

The bar is high: engineers with strong judgment who set technical direction, turn prototypes into reliable systems fast, and are at the frontier of agent-first engineering practice.

 

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

Our engineering team works with a modern tech stack designed for scalability, performance, and developer efficiency:

  • 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

Labelbox strives to ensure pay parity across the organization and discuss compensation transparently.  The expected annual base salary range for United States-based candidates is below. This range is not inclusive of any potential equity packages or additional benefits. Exact compensation varies based on a variety of factors, including skills and competencies, experience, and geographical location.

Annual base salary range
$250,000$280,000 USD

Life at Labelbox

  • Location: Join our dedicated tech hub in San Francisco
  • Work Style: Hybrid model with 3 days per week in office, combining collaboration and flexibility
  • Environment: Fast-paced and high-intensity, perfect for ambitious individuals who thrive on ownership and quick decision-making
  • Growth: Career advancement opportunities directly tied to your impact
  • Vision: Be part of building the foundation for humanity's most transformative technology

Our Vision

We believe data will remain crucial in achieving artificial general intelligence. As AI models become more sophisticated, the need for high-quality, specialized training data will only grow. Join us in developing new products and services that enable the next generation of AI breakthroughs.

Labelbox is backed by leading investors including SoftBank, Andreessen Horowitz, B Capital, Gradient Ventures, Databricks Ventures, and Kleiner Perkins. Our customers include Fortune 500 enterprises and leading AI labs.

Your Personal Data Privacy: Any personal information you provide Labelbox as a part of your application will be processed in accordance with Labelbox’s Job Applicant Privacy notice.

Any emails from Labelbox team members will originate from a @labelbox.com email address. If you encounter anything that raises suspicions during your interactions, we encourage you to exercise caution and suspend or discontinue communications.

¿Listo para postularte en Labelbox?
Postúlate en Labelbox

Cómo se compara este salario de ML Engineer

Este puesto paga $265,000/yren línea con el rango típico para los puestos de ML Engineer.

$205,000 la mediana de $240,000 $275,000

Rango típico $231,000–$275,000/yr, a partir de 11 ofertas comparables de ML Engineer en JobsRadar (salario anualizado en USD). Ver datos salariales de ML Engineer →

Empleos similares

Brain Co.
Early Career Machine Learning Engineer, Applied AI
Brain Co.
⚡ Postúlate pronto San Francisco Bay Area Presencial
● Nuevo 👁 Visto ✓ Postulado hace 1d
Brain Co.
Machine Learning Engineer, Applied AI
Brain Co.
⚡ Postúlate pronto San Francisco Bay Area Presencial
● Nuevo 👁 Visto ✓ Postulado hace 1d
Nebius
Senior Machine Learning Engineer, LLM Inference Optimization
Nebius
⚡ Postúlate pronto Palo Alto, California, United... Presencial $195,200–$262,200
● Nuevo 👁 Visto ✓ Postulado hace 2d
DigitalOcean
Staff Forward Deployed Engineer, AI/ML
DigitalOcean
⚡ Postúlate pronto San Francisco Presencial $220,000–$239,000
● Nuevo 👁 Visto ✓ Postulado hace 6d
Discord
Senior Software Engineer, Machine Learning (Ads)
Discord
⚡ Postúlate pronto San Francisco Bay Area Presencial
● Nuevo 👁 Visto ✓ Postulado hace 1sem
Liftoff
Software Engineer, ML Data
Liftoff
⚡ Postúlate pronto San Francisco Bay Area Presencial $180,000–$230,000
● Nuevo 👁 Visto ✓ Postulado hace 1sem
Ema
Software Engineer, Machine Learning
Ema
⚡ Postúlate pronto San Francisco Bay Area Presencial $160,000–$220,000
● Nuevo 👁 Visto ✓ Postulado hace 1sem
Retell AI
Senior Machine Learning Engineer
Retell AI
⚡ Postúlate pronto San Francisco Bay Area Presencial $225,000–$325,000
● Nuevo 👁 Visto ✓ Postulado hace 3sem
Kodiak
Staff Machine Learning Engineer - Data
Kodiak
⚡ Postúlate pronto San Francisco Bay Area Presencial $200,000–$265,000 🛂 Patrocinio de visado
● Nuevo 👁 Visto ✓ Postulado hace 3sem

Regístrate para recibir sugerencias adaptadas a los empleos que abres y las búsquedas que guardas.

Más empleos en Labelbox

Ver todos los empleos en Labelbox →

Postúlate ahora
🤖

Un momento — para

JobsRadar se creó para personas reales que están pasando un mal momento en su búsqueda de empleo — no para solicitudes automatizadas. Estás haciendo clic demasiado rápido y ahora estás bloqueado temporalmente.

Vuelve más tarde. Si de verdad estás buscando empleo, cuentas con nosotros — solo compórtate como una persona.

Catch your next role the second it’s posted.

Create a free account and we’ll watch the boards for you — the instant a job matches your search, it lands in your inbox or Telegram. No digging, no refreshing.

Create free account

Free forever · takes 30 seconds · already have one?

Toma ventaja en tu búsqueda de empleo.

Únete a nuestro canal de Telegram para lo que te ayuda a conseguir el puesto — referencias salariales, el pulso semanal del mercado y avisos de nuevas funciones. Sin spam, solo señal.

Únete al canal — es gratis