GrepJob
Labelbox

Staff ML Engineer, Agent Training & Environments

Labelbox
Apply
about 3 hours ago

Base Salary

$250k - $280k/yr

Responsibilities

  • Develop RL environments for agentic tasks, including task definitions and reward design.
  • Create verifiers and grading systems to assess agent performance reliably.
  • Build fine-tuning pipelines that translate evaluation signals into agent improvements.
  • Design evaluation systems to measure model and product quality through agent trajectories.
  • Establish training and serving infrastructure that meets the needs of frontier labs.

Requirements

  • Minimum of 3 years of experience shipping reliable systems.
  • Strong system and API design judgment with the ability to make architectural decisions.
  • Proficiency in Python and familiarity with the full tech stack.
  • Experience fine-tuning models for agentic tasks using SFT and RL methods.
  • Ability to debug training runs and reason about compute-economics.

Benefits

  • High-impact work environment with rapid career growth opportunities.
  • Collaborative culture focused on technical excellence and innovation.
  • Hybrid work model with flexibility and in-office collaboration.
  • Opportunity to work with leading investors and top-tier clients in AI.