# Member of Technical Staff, Search

**Company:** [Cohere](https://hotfix.jobs/companies/cohere)
**Location:** Remote
**Role:** Fullstack Engineering
**Skills:** Python, PyTorch, TensorFlow, JAX, Information Retrieval, LLMs, Tf-Serving, Xla/Mlir, C++, Go
**Posted:** 2025-05-23

> Develops state-of-the-art search models including embeddings and rerankers, focusing on semantic search techniques for accuracy and efficiency. Requires proficiency in Python, ML frameworks like PyTorch/TensorFlow/JAX, and information retrieval experience.

## Job Description

## Responsibilities
- 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.

## Requirements
- 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.

## Nice-to-haves
- Experience building training and/or evaluation datasets for practical use cases.
- Proficiency in other programming languages, such as C++ or Golang.
- Experience using large-scale distributed training strategies with GPUs.

## Similar roles

- [Staff Product Engineer](https://hotfix.jobs/jobs/796207a1-77e1-4f86-bef0-da9658cea69a) - Aurelian - Seattle, WA - $200k – $300k/yr
- [Senior Staff Software Engineer, Client Architecture](https://hotfix.jobs/jobs/29a99e15-fdc9-4209-8905-7203695bf08c) - Reddit - Remote - $233k – $326k/yr
- [Staff+ Software Engineer, Safeguards Review Tooling](https://hotfix.jobs/jobs/9232ee3d-7cd8-4e17-b2a4-affdb44459af) - Anthropic - San Francisco, CA - $320 – $485k
- [Staff Software Engineer, RCM Billing](https://hotfix.jobs/jobs/23c036cd-de44-4541-b0cd-229ea3ec9123) - Commure - Mountain View, CA - $170k – $240k/yr
- [Staff Software Engineer, RCM Core](https://hotfix.jobs/jobs/6827ff0a-f97d-4688-a1d2-2633ded89aa5) - Commure - Mountain View, CA - $170k – $240k/yr

**Apply:** https://hotfix.jobs/jobs/42d8ce32-40f5-473c-870e-10670e8be714
**Canonical:** https://hotfix.jobs/jobs/42d8ce32-40f5-473c-870e-10670e8be714