Cohere logo

Member of Technical Staff, Modeling - AI/ML

Cohere

Who are we?

Cohere is on a mission to scale intelligence to serve humanity. We are dedicated to training and deploying frontier models for developers and enterprises, enabling them to build AI systems for content generation, semantic search, and more. Our commitment to innovation drives us to obsess over the quality and impact of our work.

Why this role?

As a Member of Technical Staff, you will be at the forefront of AI and ML, designing and implementing novel research ideas and shipping state-of-the-art models to production. You will have access to one of the highest ratios of compute to engineers in the world, working in an environment that blurs the lines between engineering and research.

Responsibilities

  • Design, build, and scale AI systems for serving our users.
  • Research, implement, and experiment with ideas on our supercompute and data infrastructure.
  • Learn from and work with the best researchers in the field.

Who you are

  • Extremely strong software engineering skills.
  • Proficiency in Python and related ML frameworks such as TensorFlow, TF-Serving, JAX, and XLA/MLIR.
  • Experience writing kernels for GPUs using CUDA.
  • Experience training on TPUs.
  • Experience using large-scale distributed training strategies.
  • Familiarity with autoregressive sequence models, such as Transformers.

Perks

  • An open and inclusive culture and work environment.
  • Work closely with a team on the cutting edge of AI research.
  • Comprehensive health and dental benefits, including a separate budget to take care of your mental health.
  • 100% Parental Leave top-up for 6 months for employees based in Canada, the US, and the UK.
  • Personal enrichment benefits towards arts and culture, fitness and well-being, quality time, and workspace improvement.
  • Remote-flexible with offices in major cities and a co-working stipend.
  • Generous vacation policy with 6 weeks of vacation.

Location

This role is hybrid, based in San Francisco, CA, with flexibility regarding your location.

Benefits
Extracted with AI

  • Weekly lunch stipend
  • In-office lunches & snacks
  • Full health and dental benefits
  • Mental health budget
  • Parental Leave top-up for 6 months
  • Personal enrichment benefits
  • Co-working stipend
  • 6 weeks of vacation

Similar jobs

Last update: 23 minutes ago

Cohere logo
Cohere

Member of Technical Staff, AI and Machine Learning

Join Cohere as a Member of Technical Staff to design and scale AI systems, focusing on AI, ML, and TensorFlow.

Cohere logo
Cohere

Machine Learning Intern/Co-op (Winter 2025)

Join Cohere as a Machine Learning Intern to design and train cutting-edge AI models. Remote work, flexible, and inclusive culture.

Cohere logo
Cohere

Member of Technical Staff, Search

Join Cohere as a Member of Technical Staff, Search, to develop state-of-the-art models for information retrieval.

Cohere logo
Cohere

Software Engineer Intern/Co-op (Winter 2025)

Join Cohere as a Software Engineer Intern/Co-op for Winter 2025. Work on cutting-edge AI models and NLP products in a hybrid environment.

Huawei Nederland logo
Huawei Nederland

Senior ASR / TTS Researcher

Join Huawei's research center in Amsterdam as a Senior ASR/TTS Researcher, focusing on speech synthesis and AI.

Aiven logo
Aiven

Staff Software Engineer

Join Aiven as a Staff Software Engineer to develop cloud operations platforms using open-source technologies. Hybrid work in Berlin.

yourfirm GmbH logo
yourfirm GmbH

Senior Fullstack Developer for AI-Driven Mission Technologies

Seeking a Senior Fullstack Developer for AI-driven mission technologies, focusing on Java, JavaScript, Python, and C++. Remote work available.

Cohere logo
Cohere

Senior Data Annotator/AI Data Trainer - Coding

Join Cohere as a Senior Data Annotator/AI Data Trainer focusing on coding. Enhance AI models through data quality improvement.

Persona logo
Persona

LLM Backend Developer

Join Persona as a LLM Backend Developer, work remotely, and develop AI-driven backend systems for top startups.

FoodLabs logo
FoodLabs

Senior C++ Computer Vision Engineer

Join a cutting-edge AI-DeepTech startup in Berlin as a Senior C++ Computer Vision Engineer. Work on world-class on-device AI technology.

Cere Network logo
Cere Network

Principal AI Engineer

Join Cere Network as a Principal AI Engineer to drive AI innovation in Web3. Requires 10+ years in AI/ML, NLP, and software development.

Cohere logo
Cohere

Research Scholar in Machine Learning and NLP

Join Cohere as a Research Scholar to tackle NLP and machine learning challenges remotely.

DeepL logo
DeepL

Senior Backend Engineer C++

Join DeepL as a Senior Backend Engineer C++ to design and maintain scalable backend services using C++ and AI technologies.

Gorgias logo
Gorgias

Senior Full-Stack Engineer ReactJS/NodeJS

Join Gorgias as a Senior Full-Stack Engineer specializing in ReactJS and NodeJS, enhancing AI-powered ecommerce solutions.

dataroots logo
dataroots

Expert Machine Learning Engineer

Join Dataroots as an Expert Machine Learning Engineer to design and deliver AI-powered solutions, focusing on machine learning models.

cohere | local retail insights logo
cohere | local retail insights

Senior Full-stack Engineer - Founding Team Member

Senior Full-stack Engineer needed for a founding team role at a dynamic startup, focusing on retail industry innovations.

Huawei Nederland logo
Huawei Nederland

Information Retrieval Algorithm Engineer

Join Huawei as an Information Retrieval Algorithm Engineer to develop cutting-edge AI technologies in Amsterdam.

Catalyze Group logo
Catalyze Group

Full Stack Developer with AI and API Expertise

Join Catalyze Group as a Full Stack Developer to build AI-powered grant-writing tools. Work with React, Django, and more in Amsterdam.

Carbon13 logo
Carbon13

Cofounder - Full Stack Developer/Data Scientist for Climatech Startup

Join Carbon13 as a cofounder in climate tech, leveraging AI, data science, and software development to combat climate change.

Bitmovin logo
Bitmovin

Staff C++ Software Engineer

Join Bitmovin as a Staff C++ Engineer to lead video streaming tech innovations. Work with C++, Docker, Kubernetes in a hybrid role.

LEGALFLY logo
LEGALFLY

Back End Engineer with Node.js and TypeScript

Join LegalFly as a Back End Engineer to revolutionize legal AI with Node.js and TypeScript in a hybrid role in Ghent.

Reaktor logo
Reaktor

Lead Developer with DevOps and Functional Programming

Join Reaktor as a Lead Developer in Amsterdam, focusing on DevOps, Functional Programming, and JavaScript in a hybrid work environment.

CentML logo
CentML

Senior Software Engineer - LLM Inference

Join CentML as a Senior Software Engineer focusing on LLM Inference, leveraging AI, ML, and GPU technologies.

Poggio logo
Poggio

Senior AI Engineer

Join Poggio as a Senior AI Engineer to innovate AI systems for enterprise sales, focusing on AI capabilities and system performance.