Member of Technical Staff, Search
Remote • United States • FullTime
Posted 1y ago
Job Location
United States
Tech Stack
Remote Work Policy
Fully remote
Employment Type
FullTime
Categories
Applied AI Engineer
About the job
Cohere is seeking talented individuals to join our Search team as a Member of Technical Staff. This role focuses on developing state-of-the-art models for information retrieval, including training embedding and reranker models. You will have the opportunity to revolutionize search experiences by building intelligent, efficient, and precise search systems. Your work will advance semantic search techniques, improve accuracy and efficiency, and involve integrating novel technologies into our search infrastructure. We are looking for someone passionate about search with a strong background in information retrieval and experience working with diverse technologies and cross-functional teams.
Responsibilities
- Design, train, and improve 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 fast and stable inference.
- Collaborate with product teams to develop solutions.
- Engage in research collaborations and publish work in top-tier conferences and journals.
Requirements
- Proficiency in Python and ML frameworks like PyTorch, Tensorflow, TF-Serving, JAX, and XLA/MLIR.
- Familiarity with training and using various information retrieval models.
- Experience leveraging Large Language Models for training data or evaluation pipelines.
- Strong communication and problem-solving skills.
- 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.
Benefits
- Weekly lunch stipend of $75/£75 or equivalent.
- Full health and dental benefits, including a mental health budget.
- RRSP matching, 401K, Pension Scheme.
- 100% Parental Leave top-up for up to 6 months.
- Annual enrichment benefits (Arts & culture, fitness/wellness, quality time, workspace improvement).
- Education & learning stipend for conferences, courses, and coaching.
- 6 weeks of paid vacation (30 working days).
- Budget for traveling to other offices if remote.
- Annual company offsite.
- Daily lunch program, snacks, and social events for in-office employees.
- Co-working benefit for remote employees not near an office.
- $500 home office stipend.