Member of Technical Staff, Search

Cohere Cohere · AI Frontier · United States · Embeddings & Search

Cohere is seeking a Member of Technical Staff for their Search team to develop and improve state-of-the-art models for information retrieval, focusing on training embedding and reranker models. The role involves gathering and optimizing retrieval datasets, collaborating with serving and product teams, and engaging in research collaborations with a focus on publishing work.

What you'd actually do

  1. Design, train and improve upon cutting-edge search models.
  2. Gather high-quality retrieval datasets and optimize data pipelines for model training and evaluation.
  3. Work closely with the model serving team to ensure that inference is fast and stable.
  4. Collaborate with product teams to develop solutions.
  5. Engage in research collaborations with our partner organizations and academic affiliations and publish your work in top-tier conferences and journals.

Skills

Required

  • Python
  • PyTorch
  • Tensorflow
  • TF-Serving
  • JAX
  • XLA/MLIR
  • training and using various information retrieval models
  • Leveraging Large Language Models as part of training data or evaluation pipelines
  • information retrieval

Nice to have

  • building training and/or evaluation datasets for practical use cases
  • C++
  • Golang
  • large-scale distributed training strategies with GPUs

What the JD emphasized

  • training embedding and reranker models
  • advancing semantic search techniques
  • productionize your ideas
  • research collaborations
  • publish your work

Other signals

  • training embedding and reranker models
  • advancing semantic search techniques
  • productionize your ideas
  • research collaborations
  • publish your work