Cohere
Member of Technical Staff, Integration/RL Team (Research Engineer)
$250K–$535K
Paris +4 · Remote · Staff · Full-time · Posted Aug 19, 2025
- Location
- Paris
London
New York
Toronto
Montreal
About the job
Develop and scale software, tools, and distributed reinforcement-learning methods for LLM post-training.
What you'll do
- Design scalable, high-performing model-training software and research-support tools.
- Optimize SFT, offline preference, and RL training performance and speed.
- Research and implement ideas on cluster and data infrastructure, coordinating across engineering and scientific teams.
What they ask for
- Extremely strong software engineering skills.
- 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/speed profiling.
Nice to have
- Experience with distributed training infrastructure (Kubernetes) and frameworks (Ray).
- Hands-on post-training experience focused on scalability and performance, or ML, LLM and RL academic research.
Pay
CA$250K – CA$535K (base salary)
Tools
Python, JAX, Pytorch, XLA/MLIR, Kubernetes, Ray
AI focus
Develop and scale LLM post-training algorithms and infrastructure, especially large-scale distributed RL.
Model-written summary · the original posting is authoritative