Latest AI JobsBrowse Jobs
Cohere logo

Member of Technical Staff, Post-Training

CohereLondonModeling
Apply on company site
Salary
$250K - $535K CAD
Job type
Full-time
Work mode
Remote
Posted
13 Jun 2025
Status
Open · checked 7d ago

About this role

Role Overview

The Member of Technical Staff, Post-Training position at Cohere is a full-time opportunity based in London or available as a remote arrangement. This role suits talented machine learning professionals who want to advance state-of-the-art AI models, bridge research and production, and contribute to enterprise AI solutions.

Key Responsibilities

Design and write scalable, high-performance software dedicated to model training.
Continuously post-train models to achieve state-of-the-art performance levels.
Collaborate closely with specialist departments, such as Agentic and Code teams, to deliver comprehensive model performance.
Develop and apply techniques that elevate training cycle results across both Supervised Fine-Tuning (SFT) and Reinforcement Learning (RL) regimes.
Research, test, and implement innovative concepts utilizing supercompute and data infrastructure.
Work alongside and learn from leading researchers within the artificial intelligence field.

Requirements

Possess extremely strong software engineering capabilities.
Demonstrate proficiency using Python along with machine learning frameworks like PyTorch, JAX, and XLA/MLIR.
Bring practical experience with distributed training infrastructure such as Kubernetes, Slurm, and Ray.
Familiarity with large-scale distributed training strategies.
Practical, hands-on background in training large models at scale.
Direct experience in the post-training phase of model creation, focusing heavily on performance optimization.

Nice to Have

Authored or co-authored papers featured at top-tier venues including NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, and EMNLP.

Skills and Tools

Python
PyTorch
Kubernetes
JAX
XLA/MLIR
Slurm
Ray

Salary and Benefits

Annual salary ranging between 250,000 and 535,000 CAD.
Weekly lunch stipend of $75/£75 or local currency equivalent.
Comprehensive health and dental coverage, featuring a dedicated mental health budget.
Retirement plans including RRSP matching, 401K, and pension scheme options.
Full parental leave top-up providing up to 6 months of support for either parent.
Annual enrichment benefits covering wellness, fitness, arts, culture, workspace improvements, and continuous learning or conference budgets.
Generous paid time off of 6 weeks (30 working days).
Travel budget to visit company offices alongside an annual company offsite event.
Remote workers receive a home office setup stipend and a co-working space benefit.

About the Company

Cohere operates as a leading security-first enterprise artificial intelligence enterprise, creating advanced foundation models and end-to-end products designed to address real-world business challenges. Headquartered in Toronto with major international offices spanning London, New York City, San Francisco, Montreal, Paris, Berlin, and Seoul, the organization features a high compute-to-engineer ratio and an inclusive, globally distributed workforce.

How to Apply

Interested candidates should submit their application directly through the official company careers page.

Skills and tech stack

Questions candidates ask

Can I do this job remotely?

Yes, this position is remote-friendly, though candidates can also work from company office locations such as London, Paris, Toronto, San Francisco, or New York.

What is the salary range for this position?

The annual salary for this role is between 250,000 and 535,000 CAD.

What technical skills are required?

Applicants need strong software engineering skills, proficiency in Python, experience with ML frameworks like PyTorch and JAX, and familiarity with distributed training frameworks and infrastructure like Kubernetes, Slurm, and Ray.

How many days of paid vacation do employees receive?

Full-time employees receive 6 weeks, or 30 working days, of paid vacation.

Where should I submit my application?

All applications must be submitted through the official Cohere careers page.

Ready to apply?

The application is on the official Cohere careers page.

Apply on company site