Skip to content
← Cognition

Research Engineer, Mid-Training

Cognition · San Francisco

CHECKED 15H AGO

Applied AIonsite
Apply on the lab’s site ↗

At a glance

Highlights

  • Own mid-training decisions that sharpen base models into deep reasoning agents.
  • Drive coding, math and long-horizon reasoning gains powering Devin in deployment.
  • Research annealing schedules and synthetic data that scale across thousands of GPUs.

Ideal candidate

Empirically-driven researcher-engineer who thrives in ambiguity and cares about turning training insights into real reasoning gains for deployed agents.

Requirements

  • Deep familiarity with the LLM training pipeline end to end: pre-training data, optimization, architecture, and how mid-training and post-training interact
  • Hands-on experience with continual pre-training, annealing, or late-stage data mixing for large models
  • Strong intuition for data quality: what makes a dataset useful for training, how to filter and curate at scale, and how data mix choices compound across evals
  • Experience developing or evaluating synthetic data pipelines for capability improvement
  • Proficiency in Python and deep learning frameworks (PyTorch); comfortable debugging distributed training at scale
  • Strong fundamentals in optimization, statistics, and ML theory; able to distinguish real effects from noise, instability, and overfitting
  • A track record of original contributions: publications, open-source impact, or internal results that moved a capability frontier
  • Comfort operating in ambiguous, fast-moving environments where the problem definition is as important as the solution

Lab profile

About Cognition

Applied AI lab building end-to-end software agents, makers of Devin.

What they build

  • Devin

Known for

  • End-to-end software agents
  • Talent-dense team
More about Cognition →
Apply on the lab’s site ↗