Member of Technical Staff, Integration/RL Team (Research Engineer)
CohereParis, Ile-de-France, France • London, England, United Kingdom • Toronto, Ontario, Canada • New York, New York, United States • Montreal, Quebec, Canada
Posted August 19, 2025
Full Time
About the Company
Cohere is a leading security-first enterprise AI company specializing in cutting-edge foundation models and end-to-end AI products. We offer seamless integration, full customization, and easy-to-use solutions for businesses, ensuring data security and privacy across various deployment environments. Our focus on natural language processing, machine learning, and artificial intelligence makes us distinctive in solving real-world business problems.
Job Description
Cohere is seeking a Member of Technical Staff to join their Integration/RL team, focusing on scaling machine learning algorithms for large-scale distributed RL methods. The role involves designing and writing high-performance software, developing new tools, and collaborating with other engineering teams. The position emphasizes quality, speed, and innovation.
Requirements
- Extremely strong software engineering skills.
- Value test-driven development methods, clean code, and strive to reduce technical debts.
- Proficiency in Python and related ML frameworks such as JAX, Pytorch and/or XLA/MLIR.
- Experience using and debugging large-scale distributed training strategies (memory/speed profiling).
- [Bonus] Experience with distributed training infrastructures (Kubernetes) and associated frameworks (Ray).
- Hands-on experience with the post-training phase of model training, with a strong emphasis on scalability and performance.
Benefits
- Open and inclusive culture
- Close proximity to AI research
- Weekly lunch stipend
- Full health and dental benefits
- Parental Leave
- Personal enrichment benefits
- Remote work flexibility
