Cohere
Member of Technical Staff, Agentic Environments
$250K–$535K
Europe +2 · Remote · Staff · Full-time · Posted Aug 13, 2026
- Location
- Europe
Canada
New York
About the job
Build and improve scalable production systems, tools, and experiments for model training and research, collaborating with engineering and scientific teams.
What you'll do
- Design high-performing, scalable software for training models and develop tools that accelerate research and LLM training.
- Research, implement, and experiment with ideas using cluster and data infrastructure.
- Coordinate and collaborate with engineering and scientific teams.
What they ask for
- Extremely strong software engineering skills, with an emphasis on test-driven development and clean code.
- Proficiency in Python and related ML frameworks such as JAX, Pytorch and/or XLA/MLIR.
- Experience using and debugging large-scale distributed training strategies, including memory and speed profiling.
- Experience with distributed training infrastructure such as Kubernetes and associated frameworks such as Ray, plus experience in ML and LLM academic research.
Pay
CA$250K – CA$535K (base salary)
Tools
Python, JAX, Pytorch, XLA/MLIR, Kubernetes, Ray
AI focus
Work on frontier models, including model training, LLM optimization, and research experiments.
Model-written summary · the original posting is authoritative