Cohere

Member of Technical Staff, MLE

Cohere

full-time

Posted on:

Location Type: Hybrid

Location: London • United Kingdom

Visit company website

Explore more

✨ AI Apply
Apply

Job Level

Lead

Tech Stack

PythonTensorflow

About the role

  • Join a small, diverse team of engineers in designing, building, and scaling AI systems that underpin our suite of dev-centric enterprise products.
  • Work directly on North, Cohere’s all-in-one secure AI workspace platform. Here you will drive agent development in RAG, tool use, and language agents embedded in North.
  • Quickly research and experiment with novel ideas on our supercomputer and data infrastructure, ensuring our products remain at the forefront of the industry.
  • Collaborate with top researchers, engineers, and annotators to create and evaluate data for post-training LLMs, ensuring our products are of the highest quality and performance.
  • Engage with the latest AI and deep learning research, staying up to date with leading conferences such as NeurIPS, ICLR, and AAAI.
  • Leverage product data to understand usage patterns and identify areas for improvement, ensuring our products remain relevant and competitive.
  • Work closely with leadership to shape company strategy and goals, ensuring our product vision is aligned with our overall business objectives.

Requirements

  • Extremely strong software engineering skills.
  • Proficiency in Python and related ML frameworks such as Tensorflow, TF-Serving, JAX, and XLA/MLIR.
  • Deep experience in building and leading a product-centric organisation.
  • Direct experience working as part of a team building Large Language Models.
  • Released multiple features with several iterations.
  • Strong track record of creating and curating large-scale datasets.
  • Experience using large-scale distributed training strategies.
  • Familiarity with autoregressive sequence models, such as Transformers.
  • Ability to collaborate effectively with human annotators and cross-functional teams.
  • Paper at top-tier venues (such as NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, EMNLP).
Benefits
  • An open and inclusive culture and work environment
  • Work closely with a team on the cutting edge of AI research
  • Weekly lunch stipend, in-office lunches & snacks
  • Full health and dental benefits, including a separate budget to take care of your mental health
  • 100% Parental Leave top-up for up to 6 months
  • Personal enrichment benefits towards arts and culture, fitness and well-being, quality time, and workspace improvement
  • Remote-flexible, offices in Toronto, New York, San Francisco, London and Paris, as well as a co-working stipend
  • 6 weeks of vacation (30 working days!)
Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools
PythonTensorflowTF-ServingJAXXLAMLIRLarge Language Modelsdistributed training strategiesautoregressive sequence modelsTransformers
Soft Skills
software engineeringcollaborationleadershipproduct-centric organizationteamworkcommunicationresearchdata curationproblem-solvingstrategic alignment