jobs

Member of Technical Staff, Training Infra Engineer

Who are we?Cohere is the leading security-first enterprise AI company. We build cutting-edge foundation AI models and end-to-end products that are designed to solve real-world business problems.We’re training and deploying frontier models f

In inglese

Company

Cohere

Location

Paris; San Francisco; London; New York; Toronto; Montreal

Status

Open

Posted

2025-02-20T15:53:36.664+00:00

Cohere is hiring a training infrastructure engineer to work on the pipelines that train its frontier models. The role involves writing scalable training software, improving the setup and codebase for performance, and building tools that shorten training cycles; some work touches supercompute and data infrastructure. It asks for strong software engineering, Python with JAX, PyTorch and XLA/MLIR, distributed training experience with Kubernetes, Slurm and Ray, and hands-on large-model training at scale. Cohere calls the role remote-friendly with no location restrictions, and says engineering and research work are not strongly separated.