Member of Technical Staff, AI and Machine Learning

Join Cohere as a Member of Technical Staff

Cohere is on a mission to scale intelligence to serve humanity. We are at the forefront of AI technology, training and deploying frontier models for developers and enterprises. Our work is instrumental in the widespread adoption of AI, and we are looking for passionate individuals to join our team.

Why This Role?

As a Member of Technical Staff, you will have the opportunity to design and implement novel research ideas, ship state-of-the-art models to production, and maintain deep connections to academia. At Cohere, we do not strongly delineate between engineering and research, allowing you to contribute to both writing production code and conducting research based on your interests and organizational needs.

Key Responsibilities

• Design, build, and scale AI systems to serve our users.
• Research, implement, and experiment with ideas on our supercomputer and data infrastructure.
• Collaborate with top researchers, engineers, and annotators in the field.
• Drive agent development in Retrieval-Augmented Generation (RAG), tool use, and language agents.
• Work with internal annotators to create and evaluate data for post-training large language models (LLMs), ensuring quality and performance.

What We Are Looking For

• Extremely strong software engineering skills.
• Proficiency in Python and related ML frameworks such as TensorFlow, TF-Serving, JAX, and XLA/MLIR.
• Deep experience in AI data evaluation and management.
• 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.
• Publications in top-tier venues (e.g., NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, EMNLP).

Perks and Benefits

• 🤝 An open and inclusive culture and work environment.
• 🧑‍💻 Work closely with a team on the cutting edge of AI research.
• 🍽 Free daily lunch.
• 🦷 Full health and dental benefits, including a separate budget for mental health.
• 🎨 Personal enrichment benefits towards arts and culture, fitness, and well-being.
• 🏙 Remote-flexible, with offices in Toronto, New York, San Francisco, and London, and coworking stipends.
• ✈️ Paid vacation.

Location

This role is hybrid, with the option to work from our offices in Toronto, London, San Francisco, and New York, or remotely.

Diversity and Inclusion

Cohere values and celebrates diversity and strives to create an inclusive work environment for all. We welcome applicants from all backgrounds and are committed to providing equal opportunities. Should you require any accommodations during the recruitment process, please let us know, and we will work together to meet your needs.

Join us on our mission and shape the future of AI!