Member of Technical Staff, Training Infra Engineer at Cohere

Confirmed open

Engineer scalable training infrastructure for Cohere foundation models, bridging research and production with Python and distributed ML systems. The posting allows remote work without a stated location restriction.

Cohere

Search jobs by company: Cohere

Location: Remote
Advertised region: Paris / San Francisco / London / New York / Toronto / Montreal / France
Employment: FullTime
Work arrangement: remote
Technologies: Python, Kubernetes, PyTorch

See how to apply

Save job — free account Saved jobs

Create a free application profile — Keep your skills and projects in a reusable profile. Private by default; you choose when to share. Apply for this job directly with the employer.

By continuing, you agree to processing needed to manage your account and profile. Privacy policy

Role overview

Responsibilities

Design high-performance software for large-model training.
Improve training infrastructure and tooling to shorten model iteration cycles.
Experiment with compute and data infrastructure alongside researchers.

Qualifications

Strong Python engineering with frameworks such as JAX, PyTorch or XLA/MLIR.
Experience with distributed training infrastructure such as Kubernetes, Slurm or Ray.

Technology in this role

Python: Strong Python engineering with frameworks such as JAX, PyTorch or XLA/MLIR.
Kubernetes: Experience with distributed training infrastructure such as Kubernetes, Slurm or Ray.

About Cohere

Cohere builds foundation AI models and enterprise AI products.

Employer website

How to apply

Review the original posting and apply through the employer’s careers page.

View source and apply Apply directly

Source checked: 2026-11-08

Browse job categories

Python Kubernetes