Member of Technical Staff, Integration/RL Team (Research Engineer)

TL;DR

Member of Technical Staff, Integration/RL Team (Research Engineer): Developing and scaling machine learning algorithms and infrastructure for LLM post-training with an accent on large-scale, distributed RL methods. Focus on optimizing post-training algorithms and enhancing the global quality of the codebase.

Location: Remote (between UTC−06:00 and UTC+01:00)

Company

Cohere is a team of researchers, engineers, and designers passionate about building AI systems.

What you will do

• Design and write high-performing and scalable software for training models.
• Develop new tools to support and accelerate research and LLM training.
• Coordinate with other engineering and scientific teams to create a strong post-training ecosystem.
• Implement techniques to improve performance and speed up training cycles.
• Research and experiment with ideas on the cluster and data infrastructure.
• Collaborate with scientists, engineers, and teams.

Requirements

• Extremely strong software engineering skills.
• Proficiency in Python and related ML frameworks such as JAX, Pytorch.
• Experience with large-scale distributed training strategies.
• Bonus: Experience with distributed training infrastructures (Kubernetes).
• Bonus: Hands-on experience with the post-training phase of model training.
• Bonus: Experience in ML, LLM, and RL academic research.