Cohere

Cohere is seeking a Member of Technical Staff, Training Infra Engineer to design and write high-performant and scalable software for model training. This role will improve training setup, craft tools to speed up training cycles, and research/implement ideas on supercompute and data infrastructure. The ideal candidate has strong software engineering skills, proficiency in Python and ML frameworks like JAX/PyTorch, and experience with distributed training infrastructures and large-scale distributed training strategies.

Cohere is hiring a Member of Technical Staff for their Integration/RL Team, focusing on developing and scaling machine learning algorithms and infrastructure for LLM post-training, particularly large-scale, distributed RL methods. The role involves enhancing the post-training codebase, implementing new research tools, optimizing algorithms, and scaling distributed RL.