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