Cognition logo

Research, Mid-Training

Job Overview

Location

San Francisco Bay Area

Job Type

Full-time

Category

Data Science

Date Posted

April 9, 2026

Full Job Description

📋 Description

  • Research, Mid-Training role at Cognition focused on sharpening base model capabilities through data mix, synthetic data, annealing schedules, and context length extension to build foundational model performance for AI software engineers like Devin.
  • Day-to-day responsibilities include designing high-quality data mixtures, driving capability injection in coding and reasoning, developing synthetic data pipelines, optimizing learning rate schedules, extending context length, building evaluation metrics, and measuring scaling behavior of mid-training interventions.
  • Cognition is an applied AI lab building end-to-end software agents, known for Devin (the first AI software engineer) and Windsurf (an AI-native IDE), with a small, talent-dense team of competitive programmers, former founders, and researchers from top AI organizations.
  • The role offers the opportunity to make original contributions at the frontier of LLM training, work with large-scale GPU clusters, and directly impact deployed AI systems through rigorous empiricism and innovative research in a fast-moving, autonomous environment.

Skills & Technologies

Python
PyTorch
Mid-level
Onsite
Degree Required

Ready to Apply?

You will be redirected to an external site to apply.

About Cognition

Cognition is an applied AI lab renowned for creating Devin, the groundbreaking AI software engineer. This innovative platform is designed to automate complex software development tasks, serving enterprise clients and teams seeking to enhance efficiency in their engineering workflows. Devin's capabilities extend to areas like automating .NET framework migrations and serving as a data analyst, as demonstrated by customers like Eight Sleep. The company's credibility is underscored by its talent-dense team, whose founders boast 10 IOI gold medals and prior experience from leading AI firms such as Google DeepMind and Scale AI, positioning Cognition at the forefront of AI-driven software development.

Get more remote jobs like this

Subscribe to the weekly newsletter for similar remote roles and curated hiring updates.

Newsletter

Weekly remote jobs and featured talent.

No spam. Only curated remote roles and product updates. You can unsubscribe anytime.

Similar Opportunities

⏰ EXPIRES SOON
Bangkok, Bangkok, Thailand
Full-time
Expires Apr 25, 2026 (Soon)
JavaScript
TypeScript
React
+3 more

2 months ago

Apply
New York, New York, United States
Full-time
Expires Jun 13, 2026
Python
TensorFlow
PyTorch
+3 more

9 days ago

Apply
Ruby Labs Ltd. logo

Ruby Labs Ltd.

European Union
Full-time
Expires May 9, 2026
Go
Ruby
Remote

1 month ago

Apply
Paris, France
Full-time
Expires Jun 13, 2026
Python
Go
AWS
+5 more

8 days ago

Apply