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.

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.

Cohere logo
Cohere

Research Scholar in Machine Learning and NLP

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

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.

FutureHouse logo
FutureHouse

Software Engineer - Member of Technical Staff

Join FutureHouse as a Software Engineer to innovate AI systems for scientific research in San Francisco.

Cove logo
Cove

Applied Machine Learning / AI Engineer

Join Cove as an Applied Machine Learning / AI Engineer to innovate with AI, focusing on LLMs, ML workloads, and bespoke model training.

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.

Silimate (YC S23) logo
Silimate (YC S23)

Founding Engineer (AI/ML/LLM)

Join as a Founding Engineer to develop AI/ML solutions for chip design in San Francisco. Work on-site with a dynamic team.

Heretic logo
Heretic

Fall 2024 - Applied AI/ML Engineering Intern

Join Heretic Ventures as an AI/ML Engineering Intern to develop generative AI consumer businesses. Work with Python, TensorFlow, and PyTorch.

OpenAI logo
OpenAI

Engineering Manager, Human Data

Lead the Human Data Team at OpenAI, enhancing AI models like ChatGPT through data solutions. Hybrid work, based in San Francisco.

Anthropic logo
Anthropic

Engineering Manager, Finetuning Services

Lead Anthropic's Finetuning Services team, focusing on APIs and scalable infrastructure for LLMs.

Anthropic logo
Anthropic

Engineering Manager, Model Serving

Lead a team in building scalable infrastructure for serving large language models on cloud platforms. Manage partnerships and drive innovation.

story protocol logo
story protocol

Senior AI Engineer

Senior AI Engineer needed to integrate AI with blockchain, focusing on infrastructure and model tuning in San Francisco.

Hamming AI logo
Hamming AI

Senior Backend Engineer

Join Hamming AI as a Senior Backend Engineer to scale AI voice agent systems. Work with Next.JS, TypeScript, Python, and more.

Cantina logo
Cantina

Senior Machine Learning Engineer

Senior ML Engineer at Cantina, designing AI models for a social platform. Skills in AI, ML, NLP, Python. Remote options available.

Coherent Corp. logo
Coherent Corp.

Junior AI Engineer

Join Coherent Corp. as a Junior AI Engineer to develop AI applications using Python, Django, and Docker.

OpenAI logo
OpenAI

Tech Lead Manager, Human Data

Lead a team enhancing AI data solutions with OpenAI, focusing on safety and innovation in San Francisco.

Speak logo
Speak

Applied ML Engineer, Speech

Join Speak as an Applied ML Engineer, Speech, and revolutionize language learning with AI-driven phoneme recognition models. Based in San Francisco.

Hop logo
Hop

Machine Learning Engineer - Ads

Join as a Machine Learning Engineer focusing on Ads, developing predictive models in a hybrid role in New York.

Helm.ai logo
Helm.ai

Remote Software Engineer - Machine Learning and Cloud Infrastructure

Join Helm.ai as a Remote Software Engineer to develop ML tools, build cloud infrastructure, and work on AI technology.

OpenAI logo
OpenAI

New Products Platform Engineer

Join OpenAI as a New Products Platform Engineer to build future computing systems in a hybrid work model in San Francisco.