Research Engineer

AfterQuery; San Francisco (CA); On-site; USD 120,000 - 180,000; Full time

Benefits offered by this job

Health Insurance: Medical, Vision,/Dan

401(k) with Employer Match

Daily Meals: UberEats stipend

Monthly Wellness Stipend

Job summary

AfterQuery in San Francisco seeks a Research Engineer to design and run training experiments that quantify how datasets affect model behavior. You’ll own the end-to-end experiment loop from hypothesis to results, collaborating with partner labs and internal teams to convert data into defensible evidence.

You will build post-training infrastructure, orchestrate experiments, and ship reproducible pipelines across datasets, models, and training recipes.

Qualifications

• At least 2 years of professional experience in ML/IR or closely related field; 2-4+ years preferred.
• Strong Python and software-engineering skills.
• Hands on experience with PyTorch, JAX, Ray, and Slurm.
• Experience building production-quality ML training, evaluation, or data infrastructure.
• Hands-on experience with LLM fine-tuning, post-training, and evaluation.
• Ability to build reliable, reproducible systems for launching and comparing ML experiments.
• Strong debugging skills across distributed systems and data pipelines.
• Understanding of experimental design and deriving actionable conclusions from noisy results.

Responsibilities

• Design and build post-training infrastructure for SFT, RL and evaluation workflows.
• Build reliable pipelines for data preparation, dataset versioning, sampling, training, checkpoint management, and evaluation.
• Develop experiment orchestration and tracking systems that are reproducible and debuggable.
• Create reusable abstractions to launch experiments quickly across datasets, models, and training recipes.
• Integrate systems with partner lab training stacks, model APIs, compute environments, and evaluation infra.

Skills

Python

Distributed systems

Software engineering

Experimentation

Debugging

Education

Tools

PyTorch

JAX

Ray

Slurm

Job description

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. Forbes reported that we could be YC's fastest unicorn, reportedly raising at a $3.2 billion valuation. 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.

Why Apply

Massive Opportunity: Forbes reported that we could be YC's fastest unicorn, reportedly raising at a $3.2 billion valuation, and we're not slowing down.

Founding Impact: You will own and architect core infrastructure systems that power our platform from the ground up.

Equity & Growth: Competitive salary and meaningful equity. As we scale, you’ll have the opportunity to shape the engineering organization and lead major technical initiatives.

Strong Team: Our founding team has experience from Citadel Securities, Meta, Google, Silver Lake, and Morgan Stanley — work alongside world-class engineers and researchers.

Overview

As the Research Engineer, you will design and run training experiments that isolate the impact of our datasets on model behavior. Through controlled SFT and RL - base post-training experiments, you will measure how different data sources, structures, and selection strategies affect capability, generalization, and alignment. You will own the full experiment loop: formulate the hypothesis, build the pipeline, run the model, analyze the results, identify confounders, and determine what should happen next. Working with partner labs and internal teams, you will turn our datasets into clear, defensible evidence: this data -> this improvement -> under these conditions. This is empirical, high-leverage work for someone who likes building quickly and extracting signal from noisy results.

Responsibilities

• Design and build our post-training infrastructure for SFT, RL and evaluation workflows.
• Build reliable pipelines for data preparation, dataset versioning, sampling, training, checkpoint management, and evaluation.
• Develop experiment orchestration and tracking systems that make runs reproducible, comparable, and easy to debug.
• Create reusable abstractions that allow researchers to launch experiments quickly across datasets, models, and training recipes.
• Integrate our systems with partner-lab training stacks, model APIs, compute environments, and evaluation infrastructure.

Required Qualifications

• At least 2 years of professional experience in machine learning engineering, research engineering, ML infrastructure, or a closely related field; 2-4+ years preferred.
• Strong Python and software-engineering skills.
• Hands on experience with PyTorch, JAX, Ray, and Slurm.
• Experience building production-quality ML training, evaluation, or data infrastructure.
• Hands-on experience with LLM fine-tuning, post-training, and evaluation.
• Ability to build reliable, reproducible systems for launching and comparing ML experiments.
• Strong debugging skills across distributed systems, data pipelines, training infrastructure, and model behavior.
• Understanding of experimental design and the ability to extract actionable conclusions from noisy results.
• Ability to move quickly between infrastructure engineering and hands-on experimentation.
• A bias toward building, testing, and shipping.

Company Benefits (For Eligible Employees)

• Health Insurance: Medical, Vision, Dental
• 401(k) with Employer Match
• Daily Meals: Daily UberEats Stipend
• Monthly Wellness Stipend

We are an equal opportunity employer committed to providing a workplace free from discrimination and harassment. Employment decisions are made without regard to legally protected characteristics under applicable federal, state, or local law.

We comply with applicable pay transparency requirements and provide compensation ranges based on the position, qualifications, experience, and other relevant factors. Reasonable accommodations are available to qualified individuals with disabilities and for sincerely held religious beliefs, as required by law. This job description is intended to describe the general nature and level of work performed and is not an exhaustive list of all duties, responsibilities, qualifications, or working conditions associated with the position. We reserve the right to modify this job description as business needs change.