Member of Technical Staff, SearchActive

The opportunity

Cohere is the leading security-first enterprise AI company. We build cutting-edge foundation AI models and end-to-end products that are designed to solve real-world business problems.

What you'll do

  • 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!

What they're looking for

  • 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.