Cognition

Research Engineer, Post-Training at Cognition (San Francisco, CA)

Cognition· San Francisco, CA·

Role details

Work type
Onsite
Employment
Full-Time
Skills
Machine LearningProbabilityStatisticsLarge-scale distributed trainingRLHFRLAIFPreference modelingReward learning

Cognition at a glance

Applied AI lab behind Devin, the autonomous AI software engineer, and the Windsurf IDE.

Cognition is an applied AI lab building autonomous software engineering agents. Its flagship, Devin, plans, writes, tests, and ships production code inside a customer's own codebase and tooling. After acquiring Windsurf in 2025, it also runs the Windsurf AI-native IDE as a synchronous complement to Devin's async agent work.

$1.5B+ raised · latest: Series D · $1B+ · $26B valuation · May 2026 · backed by Lux Capital, General Catalyst, 8VC, Founders Fund

Summary

This role focuses on post-training engineering for AI agents, involving the development of training recipes, evaluation design, and alignment techniques. The candidate will iterate on model capabilities to ensure safety and effectiveness in real-world scenarios, combining deep research with hands-on engineering.

Job description

WE ARE AN APPLIED AI LAB BUILDING END-TO-END SOFTWARE AGENTS.

We're the makers of Devin, the first AI software engineer.

Our team is extremely talent-dense. Among our founding team, we have world-class competitive programmers, former founders, and leaders from companies at the cutting edge of AI including Scale AI, Palantir, Cursor, Waymo, Tesla, Lunchclub, Modal, Google DeepMind, and Nuro.

Building Devin is just the first step—our hardest challenges still lie ahead. If you’re excited to solve some of the world’s biggest problems and build AI that can reason on real-world tasks, apply to join us.

ROLE MISSION

Post-training is the critical bridge between raw model capability and a system that is actually useful, safe, and effective in the real world. You will shape how our agents learn by iterating on training recipes, evaluations, and alignment methods that directly determine what Devin and our future systems can do. This role blends deep research and hands-on engineering. We don't distinguish between the two.

WHAT YOU'LL ACCOMPLISH

  • Post-Training Recipe Development: Iterate on the full stack of datasets, training stages, and hyperparameters that determine model behavior. Measure how choices compound across evals and production performance, not just isolated benchmarks.
  • Evaluation Design and Integrity: Build evals that actually capture what matters. The loop never ends: define, optimize, realize the gaps, and rebuild. You'll be responsible for making numbers go up and making sure the numbers mean something.
  • Deep Understanding: When training produces results that don't make sense, you dig until you understand why. The goal isn't just to fix it; it's to carry that understanding forward to the next problem.
  • Alignment and Agent Behavior: Apply and advance techniques like RLHF, RLAIF, and constitutional approaches to shape how agents reason, act, and collaborate with humans in long-horizon tasks.
  • Scaling and Exploration: Measure how performance scales with data and compute, and develop new methodologies when existing ones hit ceilings. We expect both rigor and invention.

EXCEPTIONAL CANDIDATES HAVE DEMONSTRATED

  • A track record of advancing ML systems through post-training, alignment, or related methods: RLHF, RLAIF, preference modeling, reward learning, or equivalent
  • Strong fundamentals in probability, statistics, and ML theory. The ability to look at experimental data and distinguish real effects from noise and bugs
  • Evidence of original contributions: publications at top venues, open-source impact, or equivalent industry results
  • Experience with large-scale distributed training and the debugging that comes with it
  • Systems-level thinking: not just model optimization, but understanding how training pipelines, data, and evaluation interact
  • Comfort with ambiguity and fast-moving research environments where priorities shift quickly
  • We care more about demonstrated capability than credentials. A PhD is one signal among many.

RESOURCES & ENVIRONMENT

  • Small, highly selective team where research and product move together; prototypes reach real deployment quickly
  • Compute is not a constraint: large allocations with training jobs routinely running across thousands of GPUs from day one
  • The environment rewards speed, autonomy, and technical depth with minimal process overhead; this is one of the most competitive and fast-moving problems in AI
  • Everything needed to operate at frontier scale from day one.

EQUAL OPPORTUNITY

Cognition is an equal opportunity employer. We do not discriminate on the basis of race, color, religion, sex, sexual orientation, gender identity, national origin, age, disability, veteran status, or any other protected characteristic under applicable law. We are committed to providing reasonable accommodations for candidates with disabilities throughout the hiring process - please let us know if you need any.

Why work at Cognition

  • Culture: “Talent-dense,” interdisciplinary, low-ego, and urgency-driven. The team works on “the hardest problems in AI” and values shipping real impact.
  • Work model: Hybrid/office-first with major hubs in San Francisco, New York, Austin, London, Singapore, Tokyo, and others. Many roles list specific locations, suggesting in-office presence is expected.
  • Benefits: Comprehensive medical, dental, vision insurance; unlimited PTO; meals in primary offices; wellness stipend; commuter benefits; 401(k); HSA/FSA; fertility and family building support.
  • Engineering culture: Strong emphasis on applied AI, customer proximity (“we work alongside our customers”), and building infrastructure that redefines software engineering.

Application questions