# Staff Software Engineer, AI Data Platform

**Company:** [Labelbox](https://hotfix.jobs/companies/labelbox)
**Location:** San Francisco, CA
**Role:** ML Engineering
**Salary:** $250k – $280k/yr
**Experience:** 4+ years
**Skills:** TypeScript, Python, React, Node.js, GraphQL, GCP, Kubernetes, MySQL, Postgres, Kafka
**Posted:** 2026-06-09

> Staff-level engineer building AI data platform infrastructure, eval systems, and agent-first tooling for frontier labs. Requires 4+ years shipping production systems, full-stack experience, and deep TypeScript/Python proficiency.

## Job Description

## What you may work on
- Eval systems that run millions of agent trajectories to measure model and product quality.
- Fine-tuning pipelines that turn evaluation signals into measurable agent improvements.
- Agent-first product surfaces: UX and infrastructure for workflows where the user is a model or an agent operator.
- The systems behind hundreds of thousands of AI interviews used to source and match freelance workers to projects.
- Infrastructure that scales to the throughput frontier labs actually need.
- Integration of the latest models and capabilities into production within days of release.

## What we're looking for
- 4+ year track record of shipping systems customers and other engineers rely on
- You build full stack prototypes fast and they hold up. The v1 you ship becomes the foundation the rest of the team builds on.
- Strong system and API design judgement
- Hard architecture and product 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 to further accelerate the team's velocity.
- You set direction by being the example. Other engineers reach for your designs and your code as the reference.
- You move fast in ambiguous, startup-pace environments with influence over authority.
- You have worked in all parts of the stack
- Deep proficiency in TypeScript and/or Python.

## Nice to have
- Production experience building LLM- or agent-driven products.
- Designing evaluations for LLMs and agents, or producing high-quality data for ML systems.
- Background in production distributed systems, ML infrastructure, or data systems at scale.

## 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, 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
- [Staff AI Engineer - AI Product](https://hotfix.jobs/jobs/c1f2fb81-d918-4abd-873b-9b108332280f) - ClickUp - Remote - $250k – $300k/yr

**Apply:** https://hotfix.jobs/jobs/1914a72f-f09f-498a-8edf-51b951be967a
**Canonical:** https://hotfix.jobs/jobs/1914a72f-f09f-498a-8edf-51b951be967a