Become a pivotal member of Cohere's technical team, specializing in AI performance engineering. Your expertise will help in advancing our language models through optimizations and robust system designs.Cohere is looking for a Technical Staff member focusing on improving the performance of AI models. This role involves working within our Pre-Training team to enhance model training metrics while utilizing your software engineering expertise. You will contribute to the creative landscape of natural language processing from anywhere in the world, preferably aligning with ET to CET time zones for collaboration.Key Responsibilities:Create scalable and robust software for model trainingAnalyze design choices and their impacts on throughputWrite performance-enhancing CUDA and triton kernelsExplore solutions on advanced data infrastructuresEngage with leading researchers to drive innovationRequirements:Extensive software engineering skillsStrong knowledge of Python and ML frameworks, especially JAXProficient in writing GPU kernels with CUDA and tritonExperience with distributed training methodsFamiliarity with autoregressive models, particularly TransformersContribute your technical acumen to elevate AI capabilities at Cohere.

📌 Technical Staff In Ai Performance Engineering (Toronto)
🏢 Cohere
📍 Toronto