machine learning engineer for enterprise AI products

Cohere is a security-first enterprise AI company that builds foundation AI models and end-to-end products for real-world business problems. It trains and deploys frontier models for enterprises building AI systems.

• Plan and execute large-group projects from ideation through production;
• Align engineering, product, and other disciplines;
• Mentor a distributed team of engineers in subject matter expertise;
• Identify opportunities and gaps in existing models and strategize areas of focus;
• Work closely with product teams to develop solutions;
• Collaborate with partner organizations;
• Assist legal teams with preparing patents for developed IP;
• Design and implement solutions using Large Language Models for customer problems.

• Prior experience leading ML teams in applied research and production;
• Experience setting and executing technical direction for building new or refining existing products;
• Experience working across disciplines in matrixed organizations;
• Proficiency in Python and ML frameworks such as Tensorflow, TF-Serving, JAX, and XLA/MLIR;
• Experience with large-scale distributed training and inference;
• Strong mentorship, communication, and problem-solving skills;
• Demonstrated passion for applied ML models and products;
• Nice to have: experience working for B2B companies in the ML space.

• Weekly lunch stipend of $75/£75 or equivalent in local currency;
• Full health and dental benefits, including a separate mental health budget;
• RRSP matching, 401K, and Pension Scheme;
• 100% Parental leave top-up for up to 6 months for either parent;
• Annual enrichment benefits covering arts and culture, fitness and wellness, quality time, and workspace improvements;
• Education and learning stipend for conferences, courses, and coaching;
• 6 Weeks of paid vacation (30 working days);
• Budget for travel to other offices for remote employees and an annual company offsite;
• Daily lunch program, snacks, and community and social events for office employees;
• Co-working benefit for employees not near an office;
• $500 Home office stipend.