Research Engineer
AfterQuery · San Francisco, CA · Yesterday
On-siteEngineering$210k–$450k/yrFull-time
About AfterQuery
Affiliated with the Applied Research Lab, AfterQuery curates data solutions for foundation model development. Serving every frontier AI lab, we aim to deliver the best data to power the best models, making expertise accessible to all. Our customers are the builders of foundational models, and our work is integral to their model improvements.
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.
- Create experiment orchestration and tracking systems that make runs reproducible, comparable, and easy to debug.
- Develop 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
- Health Insurance: Medical, Vision, Dental
- 401(k) With Employer Match
- Daily Meals
- Daily UberEats Stipend
- Wellness Stipend
- Monthly - Covers Equinox Membership
- Commute Covered
Compensation Range
$210K - $450K