# Member of Technical Staff, Post-Training

**Company**: Cohere
**Location**: London; Montreal; New York; Paris; San Francisco; Toronto
**Work arrangement**: hybrid
**Experience**: senior
**Job type**: Full time
**Salary**: CA$250K - CA$535K
**Category**: Engineering
**Industry**: Technology

**Apply**: https://jobs.ashbyhq.com/cohere/554a9380-ab50-4338-88a9-c6b8ab19d92e?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply
**Canonical**: https://yubhub.co/jobs/job_0e49d25a-fc4

## Description

Cohere is committed to fair and transparent pay practices. The compensation range listed for this role reflects the expected compensation. Actual compensation offered will be determined by factors such as location, level, job-related knowledge, skills, education, and experience.

Advance the state of the art for model post-training, ship state-of-the-art models to production, and bridge the gap between research and production. You will have one of the highest ratios of compute to engineers in the world. You will not be confined to traditional engineering or research roles; instead, you will contribute to writing production code and supporting our research efforts based on your interests and organisational needs.

**Key Responsibilities:**

- Design and write high-performant and scalable software for training models.

- Consistently post-train models to reach state-of-the-art level performance.

- Coordinate with other specialist teams to produce models with strong overall performance.

- Craft and implement techniques to improve the performance and results of our training cycles.

- Research, implement, and experiment with ideas on our supercompute and data infrastructure.

- Learn from and work with top researchers in the field.

**Qualifications:**

- Strong software engineering skills.

- Proficiency in Python and related ML frameworks such as JAX, PyTorch, and XLA/MLIR.

- Experience with distributed training infrastructures and associated frameworks.

- Experience using large-scale distributed training strategies.

- Hands-on experience training large models at scale.

- Hands-on experience with the post-training phase of model training, focusing on performance optimisation.

**Benefits:**

- A weekly lunch stipend of $75/£75 or equivalent in your 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.

- Annual enrichment benefits for arts & culture, fitness/wellness, and education.

- 6 weeks of paid vacation.

- Budget for traveling to other offices and an annual company offsite.

**Working Location:** This role can be based remotely or from one of our office locations in London, Paris, Toronto, San Francisco, or New York. We are remote-friendly and care about hiring exceptional people regardless of location.

## Skills

### Required
- Python
- JAX
- PyTorch
- XLA/MLIR
- distributed training infrastructures
- large-scale distributed training strategies

### Nice to have
- paper at top-tier venues

---

Source: [Apply at jobs.ashbyhq.com](https://jobs.ashbyhq.com/cohere/554a9380-ab50-4338-88a9-c6b8ab19d92e?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply)