Open role
Research Engineer, Mid-Training
Cognition
San Francisco, California, United StatesPosted Aug 6, 2026 · 8h ago
Full-timeMid LevelOn-siteAI / ML
About this role
Join an applied AI lab as a Research Scientist focused on mid-training for AI software agents like Devin. You will own critical late-stage training decisions, influencing model capabilities in coding, math, and reasoning. This role requires a blend of research and engineering expertise in a fast-paced, high-compute environment.
What we are looking for
6- Own late-stage training decisions for AI models
- Design data mixtures and quality uplift strategies
- Drive capability injection in coding, math, and reasoning
- Develop and evaluate synthetic data pipelines
- Research and optimize annealing and schedule design
- Extend context length without performance degradation
