Technical Staff Member, Search

Cohere

full-time

Posted on: 3/18/2026

Location Type: Remote

✨ AI Apply

About the role

Design, train and improve upon cutting-edge search models.
Gather high-quality retrieval datasets and optimize data pipelines for model training and evaluation.
Work closely with the model serving team to ensure that inference is fast and stable.
Collaborate with product teams to develop solutions.
Engage in research collaborations with our partner organizations and academic affiliations and publish your work in top-tier conferences and journals.
Join us at a pivotal moment, shape what we build, have a strong ownership mindset, and wear multiple hats!

Proficiency in Python and related ML frameworks such as PyTorch, Tensorflow, TF-Serving, JAX, and XLA/MLIR.
Familiarity with training and using various information retrieval models.
Experience leveraging Large Language Models as part of training data or evaluation pipelines.
Strong communication and problem-solving skills.

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

PythonPyTorchTensorFlowTF-ServingJAXXLAMLIRinformation retrieval modelsLarge Language Modelsdata pipelines

Soft Skills

communicationproblem-solvingownership mindset