Software Engineer - RL Environments

Posted 20 Days Ago
San Francisco, CA, USA
In-Office
200K-200K Annually
Junior
Artificial Intelligence • Big Data
The Role
Design datasets, evaluation rubrics, and reward signals for RLHF/RLVR; build real and synthetic data pipelines; run experiments modeling annotator behavior; develop quantitative metrics for dataset quality, diversity, and downstream impact; partner with research teams to translate training objectives into data and evaluation specs.
Summary Generated by Built In
About AfterQuery

AfterQuery is an applied research lab curating data solutions for foundation model development.

We serve every frontier AI lab with the mission of delivering the best data to power the best models. In doing so, we can make expertise that once took a lifetime to build available to anyone who needs it. Our customers are the ones building the foundation models themselves and our work sits directly in the loop of how those systems improve.

This is a rare opportunity to join a company at a defining moment in AI. Since raising our $30M Series A at a $300M valuation, AfterQuery has grown well over a $100M revenue run rate.

We're based in San Francisco and backed by leading investors including Altos Ventures, BoxGroup, and Y Combinator and angels from Google DeepMind, OpenAI, Anthropic, Meta Superintelligence Labs, and Microsoft AI.

The Role

As a SWE (Environments), you will design the datasets and evaluation rubrics that directly influence how frontier models learn. You'll work hands-on with research teams at top AI labs, experimenting with data collection strategies, diagnosing model failure modes, and developing the metrics that determine whether a model is actually improving. You'll go from hypothesis to live experiment quickly, and your output will feed directly into model training runs at scale.

Day to day, you will design data slices that expose meaningful failure modes across domains like finance, code, and enterprise workflows. You will build and refine reward signals for RLHF and RLVR pipelines. You will develop quantitative frameworks for measuring dataset quality, diversity, and downstream impact on alignment and capability. You will partner with lab research teams to translate their training objectives into concrete data and evaluation specifications.

What You'll Do

  • Design data slides and explore data shapes that expose meaningful model failure modes across domains like finance, code, and enterprise workflows

  • Build and refine evaluation rubrics and reward signals for RLHF and RLVR training pipelines

  • Model annotator behavior and run experiments to improve different model capabilities

  • Develop quantitative frameworks for measuring dataset quality, diversity, and downstream impact on model alignment and capability

  • Create and manage both real world & synthetic data pipelines

  • Partner with lab research teams to translate their training objectives into concrete data and evaluation specifications

What We're Looking For

  • 1-4 YOE

  • Major plus if they've worked for/interned for any RL environment companies in the past or any AI safety or benchmarking orgs like METR, Artificial Analysis, etc..

  • Genuine obsession with how data structure, selection, and quality drive model behavior

  • Ability to design lightweight experiments, move fast, and extract actionable insights from messy results

  • Former founders and early engineers at early stage startups are a plus. We don't filter on pedigree. We want people who can demonstrate they work hard, learn fast, and care deeply about getting the details right.

Compensation Structure:

$200k base + profit share (around 150% of base) + competitive equity

Skills Required

  • 1-4 years of experience
  • Experience designing datasets, data slices, and evaluation rubrics
  • Experience building and refining reward signals for RLHF and RLVR pipelines
  • Experience creating and managing real-world and synthetic data pipelines
  • Experience modeling annotator behavior and running experiments to improve model capabilities
  • Ability to design lightweight experiments, move fast, and extract actionable insights from messy results
  • Genuine focus on data structure, selection, and quality driving model behavior
  • Experience at RL environment companies or AI safety/benchmarking organizations (e.g., METR, Artificial Analysis)
  • Former founder or early engineer at an early-stage startup
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
200 Employees

What We Do

AfterQuery is an applied research lab curating data solutions to accelerate foundation model development.

Similar Jobs

Optum Logo Optum

Licensed Vocational Nurse

Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
In-Office
Mission Hills, CA, USA
160000 Employees
20-36 Hourly

Micron Technology Logo Micron Technology

Design Engineer

Artificial Intelligence • Hardware • Information Technology • Machine Learning
In-Office
2 Locations
45000 Employees
177K-334K Annually

Micron Technology Logo Micron Technology

VP, FP&A

Artificial Intelligence • Hardware • Information Technology • Machine Learning
In-Office
2 Locations
45000 Employees
232K-393K Annually

Micron Technology Logo Micron Technology

Principal Engineer

Artificial Intelligence • Hardware • Information Technology • Machine Learning
In-Office
3 Locations
45000 Employees
161K-318K Annually

Similar Companies Hiring

Legora Thumbnail
Artificial Intelligence • Legal Tech • Software
Chicago, Illinois
700 Employees
Hanover Park Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
42 Employees
Onshore Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
60 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account