Member of Technical Staff - Post-training (Data Infrastructure)

The role

You will build and own pipelines that produce post-training data for large language models. This includes SFT, RL, synthetic, and human-annotated data. You will work with researchers and annotation teams to iterate on data.

What you will do

• Build and maintain data pipelines for SFT, preference, and RL post-training data.
• Own dataset quality, versioning, and reproducibility.
• Turn research data recipes into production-grade, reusable components.
• Scale model inference for data generation and model-based grading.
• Design data formats and schemas for post-training datasets.
• Collaborate with researchers and engineers to keep data pipelines aligned with model needs.

Required qualifications

• Strong software engineering skills.
• Experience designing and building large-scale data pipelines for machine learning, ideally for LLM post-training.
• Track record of running distributed data processing and model inference at scale.
• Good understanding of post-training methods (e.g., SFT, RLHF, DPO) and how data shapes model behaviour.
• Excellent communication skills.

Benefits

• Weekly lunch stipend of $75/£75 or equivalent in local currency.
• Full health and dental benefits, including a separate budget for mental health.
• RRSP matching, 401K, Pension Scheme.
• 100% Parental Leave top-up for up to 6 months, for either parent.
• Annual enrichment benefits for arts & culture, fitness/wellness, quality time, and a workspace improvement credit.
• Education & learning stipend for conferences, courses, and coaching.
• 6 weeks of paid vacation (30 working days!).
• Budget for traveling to other offices if you are remote, plus an annual company offsite.
• Daily lunch program, snacks, and regular community and social events for those in the office.
• Co-working benefit for those not near an office.
• $500 home office stipend.

Before you apply

• May use AI-enabled tools to screen and assess applicants.

About the company

Cohere is a security-first enterprise AI company that builds foundation AI models and end-to-end products. It is a global technology company headquartered in Toronto.