Member of Technical Staff, Cohere

Post Training

Salary not provided

Kubernetes

Python

PyTorch

Senior and Expert level

London

New York

Paris

San Francisco Bay Area

Toronto

Remote from Canada, Europe, UK, US

Enterprise-grade AI solutions for real-world business problems

Open for applications

Enterprise-grade AI solutions for real-world business problems

501-1000 employees

B2BArtificial IntelligenceMachine LearningSaaS

Open for applications

Salary not provided

Kubernetes

Python

PyTorch

Senior and Expert level

London

New York

Paris

San Francisco Bay Area

Toronto

Remote from Canada, Europe, UK, US

501-1000 employees

B2BArtificial IntelligenceMachine LearningSaaS

Company mission

To empower businesses to scale innovation, boost productivity, and drive progress that reaches everyone.

Job

Company

Role

Who you are

Extremely strong software engineering skills
Proficiency in Python and related ML frameworks such as JAX, Pytorch and XLA/MLIR
Experience with distributed training infrastructures (Kubernetes, Slurm) and associated frameworks (Ray)
Experience using large-scale distributed training strategies
Hands on experience on training large model at scale
Hands on experience with the post training phase of model training, with a strong emphasis on performance optimisation
Bonus: paper at top-tier venues (such as NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, EMNLP)

What the job involves

Design and write high-performant and scalable software for training models
Consistently post-train the models to reach SOTA level performance
Coordinate with other specialist teams (Agentic, Code…) to produce models that have strong all encompassing performance
Craft and implement techniques to improve the performance and results of our training cycles both on the SFT and the RL regime
Research, implement, and experiment with ideas on our supercompute and data infrastructure
Learn from and work with the best researchers in the field

Our take

Many businesses are now faced with the challenge of adopting advanced AI safely and securely. While AI tools like ChatGPT are making headlines, companies (especially those in regulated industries such as banking, telecoms, or government services) worry about privacy, compliance, and control over their sensitive data. Cohere addresses these concerns by offering AI solutions that can be trained on a company's internal data and, importantly, run on their own infrastructure rather than relying on external cloud providers. And in a world where data breaches and regulatory scrutiny are constant threats, this focus on trust and security is key.

In a competitive landscape, dominated by giants like OpenAI, Google, and Elon Musk;'s xAI, Cohere is setting itself apart by targeting businesses exclusively. Its approach has earned it contracts with some big names, like Royal Bank of Canada, Fujitsu, LG, Oracle, and Dell. By focusing on reliability and enterprise adoption rather than viral consumer appeal, Cohere has carved out a strong niche for itself.

This isn't going unnoticed either, as the company recently raised $500M, boosting its valuation to $6.8B, and brought on board top talent from MEta and Uber to strengthen its leadership team. Alongside this, its doubled its annual recurring reveneu to $100M in 2025 and is aiming to hit $200M by year-end. With its North platform for AI agents and a lean cost structure compared with larger rivals, Cohere is very much ready for more growth.

Steph

Company Specialist at Welcome to the Jungle

Insights

Top investors

192% employee growth in 12 months

Glassdoor (4)

Company

Funding (last 2 of 7 rounds)

Sep 2025

$100m

LATE VC

Aug 2025

$500m

LATE VC

Total funding: $1.5bn

Company benefits

Six weeks’ paid vacation
Equity / stock options
RRSP, 401(k), and Pension Scheme contributions
Coverage for 100% of your insurance premiums across health, dental, vision, and travel
Additional coverage for accessing mental health providers/services
Six months of fully paid parental leave, including adoption and surrogacy
Financial support for egg freezing and IVF in Canada and the UK
A monthly fitness and wellness allowance
Globally dispersed company that supports a remote work culture
A $2,000 annual education benefit for professional development
A weekly stipend for meals when working remotely and catered lunch when working from one of our global offices
A monthly arts and culture allowance
A monthly quality time allowance

Company HQ

Grange Park, Toronto, ON

Leadership

Aidan Gomez

(Co-Founder & CEO)

Aidan received a PhD in Computer Science from the University of Oxford in 2024 and has served as Cohere's CEO since co-founding the company in 2019.

Share this job

View 64 more jobs at Cohere