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.