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

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.

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.

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.

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.

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.

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.

Together AI logo
Together AI

Senior Backend Engineer - Java, Rust, Go

Join Together AI as a Senior Backend Engineer in Amsterdam. Work with Java, Rust, and Go to build scalable backend systems.

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.

Nebius AI logo
Nebius AI

Senior Software Engineer (C++)

Join Nebius as a Senior Software Engineer (C++) to develop reliable cloud services in a hybrid work environment.

Personio logo
Personio

Staff Software Engineer, Data Platform

Join Personio as a Staff Software Engineer in Berlin to build scalable data platforms using Kafka, Kubernetes, and AWS. Drive innovation and excellence.

HeyJobs logo
HeyJobs

Senior Software Engineer - AWS, Python, Ruby on Rails

Join HeyJobs as a Senior Software Engineer to design scalable systems using AWS, Python, and Ruby on Rails in a dynamic team.

Computer Futures logo
Computer Futures

Cloud Data Engineer

Seeking a Cloud Data Engineer with expertise in AWS, Python, and CI/CD for a hybrid role in Hannover. Join our dynamic team!

Nebius AI logo
Nebius AI

Senior Backend Engineer (Go)

Join Nebius as a Senior Backend Engineer (Go) to develop fault-tolerant cloud services in a hybrid work environment.

Bitmovin logo
Bitmovin

Senior C++ Software Engineer

Join Bitmovin as a Senior C++ Software Engineer to develop scalable video streaming solutions using modern C++ and cloud-native architectures.

DwellFi  logo
DwellFi

AI Solutions Software Engineer

Join DwellFi as an AI Solutions Software Engineer to develop innovative AI solutions using LangChain or Llama. Remote position in Palo Alto, CA.

Computer Futures logo
Computer Futures

Mid-Level Full Stack Software Engineer - Cloud & Web

Join as a Full Stack Software Engineer focusing on C#, Azure, and Microservices in a dynamic team with flexible work options.

BCG X logo
BCG X

AI Engineer

Join BCG X as an AI Engineer in Milan, Italy. Develop AI solutions, partner with clients, and drive innovation in a dynamic environment.

Motius logo
Motius

Senior Backend Developer

Join Motius as a Senior Backend Developer to work on cutting-edge R&D projects using AWS, Docker, GraphQL, and more in a hybrid work environment.

Delivery Hero logo
Delivery Hero

Software Engineer (JavaScript/React) - Display Ads (AdTech)

Join Delivery Hero as a Software Engineer (JavaScript/React) in Berlin to develop cutting-edge AdTech solutions for the local delivery market.

TrueLayer logo
TrueLayer

Senior Software Engineer - C#/.NET

Join TrueLayer as a Senior Software Engineer in Milan, working with C#, .NET, AWS, and Kubernetes to build scalable systems.

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.

Haufe Akademie logo
Haufe Akademie

Software Engineer AWS & TypeScript

Join Haufe Akademie as a Software Engineer specializing in AWS & TypeScript to develop innovative cloud solutions in Freiburg.

Braintrust logo
Braintrust

iOS Developer

Join Tezza as an iOS Developer to enhance our app with Swift and SwiftUI, working remotely in a creative team.