AfterQuery
Research Engineer
$210K–$450K
San Francisco · On-site · Full-time · Posted Aug 27, 2026
- Experience
- At least 2 years; 2-4+ years preferred.
About the job
Designs and runs post-training experiments and builds infrastructure to assess how datasets affect model capabilities, generalization, and alignment.
What you'll do
- Build post-training infrastructure and reliable pipelines for data preparation, training, checkpoint management, and evaluation.
- Develop reproducible experiment orchestration, tracking, and reusable tools for research teams.
- Integrate systems with partner-lab training stacks, model APIs, compute environments, and evaluation infrastructure.
What they ask for
- At least 2 years of professional experience in machine learning engineering, research engineering, ML infrastructure, or a closely related field.
- Strong Python and software-engineering skills.
- Hands-on experience with PyTorch, JAX, Ray, and Slurm.
- Experience building production-quality ML infrastructure and hands-on experience with LLM fine-tuning, post-training, and evaluation.
Nice to have
- 2-4+ years of experience.
Pay
$210K – $450K (base salary); $250k-450k (total compensation)
Tools
Python, PyTorch, JAX, Ray, Slurm
AI focus
Runs controlled SFT and RL post-training experiments to measure dataset effects on model behavior, capability, generalization, and alignment.
Model-written summary · the original posting is authoritative