Member of Technical Staff, Search

Stack

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 with partner organizations and academic affiliations, publishing in top-tier venues

Requirements

• ▹Proficiency in Python and ML frameworks such as PyTorch, TensorFlow, TF-Serving, JAX and XLA/MLIR
• ▹Familiarity with training and using various information retrieval models
• ▹Experience leveraging LLMs as part of training data or evaluation pipelines
• ▹Strong communication and problem-solving skills

Nice to have

• ▹Experience building training and/or evaluation datasets for practical use cases
• ▹Proficiency in other programming languages such as C++ or Go
• ▹Experience using large-scale distributed training strategies with GPUs

Soft skills

What we offer

• ▹Weekly lunch stipend ($75/£75 or local equivalent)
• ▹Full health and dental benefits, including a mental health budget
• ▹RRSP matching, 401(k) or pension scheme
• ▹100% parental leave top-up for up to 6 months, for either parent
• ▹Annual enrichment benefits (arts & culture, fitness/wellness, workspace improvement credit)
• ▹Education & learning stipend for conferences, courses and coaching
• ▹6 weeks of paid vacation (30 working days)
• ▹Travel budget to other offices plus an annual company offsite
• ▹$500 home office stipend and co-working benefit for remote employees

About the company

Cohere is a leading security-first enterprise AI company, building frontier foundation models and end-to-end AI products for real-world business problems. It is a global team co-headquartered in Toronto and San Francisco, with offices in London, New York City, Montreal, Seoul, Germany and Paris.