Skip to main content
CareerApp

Research Engineer, Mid-Training

Cognition AI

San Francisco, CA · Mid

Listed on Cognition AI’s own careers site. You apply with them directly — we never stand between you and the employer.

What this role is

This role focuses on optimizing the critical middle phase of LLM training, where raw capability is shaped into reasoning, generalization, and reliability. It suits researchers and engineers who excel at designing data strategies, training schedules, and synthetic pipelines—and who thrive making high-leverage decisions with minimal constraints.

Our summary, not Cognition AI’s wording. The full posting is on their site.

Skills this role names

Log in to see which of these are already on your profile.

What they ask for

Required

  • End-to-end familiarity with LLM training pipeline including pre-training, optimization, and architecture
  • Hands-on experience with continual pre-training, annealing, or late-stage data mixing for large models
  • Strong understanding of data quality and curation at scale
  • Experience developing or evaluating synthetic data pipelines
  • Proficiency in Python and deep learning frameworks like PyTorch
  • Ability to debug distributed training systems at scale
  • Strong fundamentals in optimization, statistics, and ML theory

Nice to have

  • Track record of original contributions through publications, open-source work, or internal results
  • Experience extending context length in language models
  • Familiarity with positional encoding strategies
  • PhD or equivalent advanced training

Turn on analytics and we load Google Analytics: Google gets the pages you open and what you do here — searches, jobs you view, jobs you apply to — and sets its own cookies. Leave it off and the only cookies we set are your login, your theme, and this answer. Accept All also records a yes to advertising, which nothing uses yet. Privacy Policy.