Member of Technical Staff, Training Performance Engineer

Join Cohere's mission to scale intelligence and serve humanity as a Performance Engineer in the Pre-Training team, optimizing language model performance and working with cutting-edge technology.

Requirements

• Extremely strong software engineering skills
• Proficiency in Python and related ML frameworks
• Experience writing kernels for GPUs using CUDA, triton, etc
• Experience using large-scale distributed training strategies
• Familiarity with autoregressive sequence models, such as Transformers

Benefits

• Open and inclusive culture
• Weekly lunch stipend, in-office lunches & snacks
• Full health and dental benefits
• 100% Parental Leave top-up for 6 months
• Personal enrichment benefits
• Remote-flexible work options
• 6 weeks of vacation