Latest AI JobsBrowse Jobs
Cohere logo

Member of Technical Staff, Training Infra Engineer

CohereParisModeling
Apply on company site
Salary
$250K - $535K CAD
Job type
Full-time
Work mode
Remote
Posted
20 Feb 2025
Status
Open · checked 8d ago

About this role

The Member of Technical Staff, Training Infra Engineer position at Cohere is a full-time role based in Paris or available remotely worldwide, tailored for software professionals eager to build advanced enterprise AI training systems. This position places the engineer at the intersection of machine learning research and high-performance production infrastructure.

Role Overview

This position involves contributing to and supporting model training pipelines while bridging the gap between research and production environments. Team members write production code and support research efforts based on organizational needs and individual interests. The organization maintains a high ratio of compute to engineers, providing access to extensive data, compute power, and talent.

Key Responsibilities

Design and develop high-performing, scalable software solutions dedicated to model training.
Enhance the existing training setup by focusing on codebase performance and infrastructure improvements.
Create and execute tools designed to accelerate training cycles and boost the overall efficiency of the training infrastructure.
Experiment with, research, and implement new ideas utilizing the supercompute and data infrastructure.
Collaborate closely with top-tier researchers in the artificial intelligence field.

Requirements

Possess extremely strong software engineering capabilities.
Demonstrate proficiency in Python alongside machine learning frameworks like PyTorch, JAX, and XLA/MLIR.
Bring practical experience working with distributed training infrastructure such as Kubernetes, Slurm, and Ray.
Show direct background in utilizing large-scale distributed training strategies.
Offer hands-on experience in training large models at scale and contributing to training infrastructure setup or tooling.

Nice to Have

Authored or co-authored a paper published at top-tier academic venues including NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, or EMNLP.

Skills and Tools

Python
PyTorch
JAX
XLA/MLIR
Kubernetes
Slurm
Ray

Salary and Benefits

Annual salary ranging from 250,000 to 535,000 CAD.
Weekly lunch stipend equivalent to 75/£75 in local currency, plus daily lunch programs for office-based employees.
Comprehensive health and dental coverage featuring a separate mental health budget.
Retirement plans including RRSP matching, 401K, and pension schemes.
Fully paid parental leave top-up of up to six months for either parent.
Annual enrichment benefits covering wellness, fitness, arts, culture, and workspace improvements, plus education and learning stipends.
Thirty working days (six weeks) of paid vacation.
Travel budget for remote staff to visit company offices and attend an annual company offsite.
Home office setup stipend of $500 or co-working space benefits for remote workers.

About the Company

Cohere operates as a security-first enterprise artificial intelligence company building foundation models and end-to-end products to resolve practical business challenges. Headquartered in Toronto with international offices spanning London, New York City, San Francisco, Montreal, Paris, Berlin, and Seoul, the organization brings together designers, researchers, and engineers focused on advancing machine learning adoption.

How to Apply

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

Skills and tech stack

Questions candidates ask

Can I work remotely for this role?

Yes, this role is remote-friendly with no location restrictions, though candidates should align with the core time zone or working hours expected for the position.

What technical skills are required?

Candidates need extremely strong software engineering skills, proficiency in Python and ML frameworks like PyTorch, and experience with distributed training infrastructures such as Kubernetes, Slurm, and Ray.

What is the salary range for this position?

The salary ranges from 250,000 to 535,000 CAD per year.

What benefits does Cohere offer full-time employees?

Perks include health and dental benefits, retirement matching, 30 days of paid vacation, parental leave top-up, weekly lunch stipends, learning budgets, and home office or co-working stipends.

Where are Cohere's physical offices located?

Cohere has corporate offices in Toronto, London, New York City, San Francisco, Montreal, Paris, Berlin, and Seoul.

Ready to apply?

The application is on the official Cohere careers page.

Apply on company site