Research Engineer, Mid-Training

About Windsurf

Cognition Technologies is an applied AI lab building end-to-end software agents.

What you'd do

• Design high-quality data mixtures for training
• Drive improvements in coding, math, and reasoning
• Develop synthetic data pipelines
• Optimize learning rate schedules and compute allocation

What they're looking for

• Familiarity with LLM training pipeline
• Experience with pre-training and late-stage data mixing
• Strong intuition for data quality
• Proficiency in Python and deep learning frameworks

Perks

• Small, selective team
• Large compute allocations for training
• Quick prototype deployment