Cohere

Cohere is seeking a Member of Technical Staff for their Safety for Agents team. This role focuses on developing better, fairer, and more trustworthy LLMs by working on data generation, post-training algorithms, and evaluation methods for AI safety. The position involves collaboration with ML, data annotation, product, and policy teams, and requires strong statistical, software engineering, and data analysis skills. The ideal candidate will have experience training LLMs and evaluating their generalizability and robustness, with a publication record in top-tier venues.

This role focuses on advancing model post-training techniques to achieve state-of-the-art performance, bridging research and production. Responsibilities include designing and writing high-performance software for training, coordinating with specialist teams, crafting techniques for SFT and RL regimes, and researching/implementing ideas on infrastructure. The role requires strong software engineering skills, proficiency in Python and ML frameworks, and experience with distributed training infrastructures and large-scale model training, particularly in the post-training phase with performance optimization.