Jobs · Engineering

Member of Technical Staff, Search

Cohere · San Francisco, CA · 1 wk ago
RemoteRemoteEngineeringFull-time

About the role

We are seeking talented individuals to contribute to the development of state-of-the-art models for information retrieval as part of our Search team. This role involves designing, training, and improving cutting-edge search models, working with a wide range of novel technologies, and collaborating with other teams to integrate your work into our search infrastructure.

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.

Qualifications

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

Skills

  • Passion for search and a strong background in information retrieval.
  • Ability to work collaboratively with other teams.
  • Interest in research collaborations and publishing work in top-tier conferences and journals.

Benefits

  • A weekly lunch stipend of $75/£75 or equivalent in your local currency for lunch.
  • Full health and dental benefits, including a separate budget for mental health.
  • RRSP matching, 401K, Pension Scheme.
  • 100% Parental Leave top-up for up to 6 months, for either parent.
  • Annual enrichment benefits: Arts & culture, fitness/wellness, quality time, and a workspace improvement credit.
  • Education & learning stipend for conferences, courses, and coaching.
  • 6 weeks of paid vacation (30 working days!).
  • Budget for traveling to other offices if you are remote, plus an annual company offsite.

Pay

  • Competitive salary based on experience and qualifications.

Schedule

  • Full-time position.

Similar jobs