At a Glance
- Tasks: Join our team to enhance AI performance and optimize language models.
- Company: Cohere, a leader in natural language processing innovation.
- Benefits: Work remotely, competitive salary, and opportunities for professional growth.
- Other info: Flexible work environment with a focus on innovation and career advancement.
- Why this job: Make a real impact in AI while collaborating with top researchers.
- Qualifications: Strong software engineering skills and experience with Python and ML frameworks.
The predicted salary is between 123750 - 151250 $ per year.
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 innovative 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 training
- Analyze design choices and their impacts on throughput
- Write performance-enhancing CUDA and triton kernels
- Explore solutions on advanced data infrastructures
- Engage with leading researchers to drive innovation
Requirements
- Extensive software engineering skills
- Strong knowledge of Python and ML frameworks, especially JAX
- Proficient in writing GPU kernels with CUDA and triton
- Experience with distributed training methods
- Familiarity with autoregressive models, particularly Transformers
- Contribute your technical acumen to elevate AI capabilities at Cohere. #J-18808-Ljbffr
Technical Staff in AI Performance Engineering in Toronto employer: Cohere
Cohere is an exceptional employer that fosters a dynamic and collaborative work culture, making it an ideal place for an Executive Assistant to thrive. With a strong focus on employee growth, we offer generous benefits including a weekly lunch stipend, comprehensive health coverage, and a robust education stipend, all while working in a cutting-edge AI environment in Toronto. Join us to be part of a passionate team dedicated to innovation and meaningful impact in the world of AI.
