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.