Cohere
Member of Technical Staff, Training Infra Engineer
$250K–$535K
Paris +5 · Remote · Staff · Full-time · Posted Feb 20, 2025
- Location
- Paris
San Francisco
London
New York
Toronto
Montreal - Experience
- Experience is requested; no number of years stated.
About the job
Build and improve scalable training infrastructure and pipelines, helping move frontier AI models from research into production.
What you'll do
- Design and write high-performance, scalable training software and improve infrastructure and codebase performance.
- Develop tools to accelerate training cycles and improve training infrastructure effectiveness.
- Research, implement, and test ideas using supercompute and data infrastructure.
What they ask for
- Extremely strong software engineering skills.
- Proficiency in Python and ML frameworks such as JAX, Pytorch, and XLA/MLIR.
- Experience with distributed training infrastructure and frameworks, including Kubernetes, Slurm, and Ray.
- Hands-on experience training large models at scale and contributing to training infrastructure tooling or setup.
Nice to have
- Papers at top-tier venues, such as NeurIPS, ICML, or ICLR.
Pay
CA$250K – CA$535K (base salary)
Tools
Python, JAX, Pytorch, XLA/MLIR, Kubernetes, Slurm, Ray
AI focus
Develop and operate infrastructure and pipelines for training large AI models and shipping them to production.
Model-written summary · the original posting is authoritative