Staff Data Scientist — LLM Data Analysis & Evaluation

Cohere; New York, San Francisco (NY, CA); Hybrid; USD 150,000 - 210,000; Full time

Benefits offered by this job

Lunch stipend

Health benefits

RRSP matching

Parental leave

Enrichment benefits

Learning stipend

Paid vacation

Travel budget

Home office stipend

Job summary

Cohere, a leading security-first enterprise AI company, is hiring a Member of Technical Staff in Data Analysis and Evaluation. You will design data-collection tasks, apply statistical methods, and assess model robustness.

Collaborate with researchers and engineers to improve dataset quality and model performance, including distributed training of LLMs. The role emphasizes rigorous experimentation, cross-functional teamwork, and contributing to high-quality AI systems that scale across diverse

Qualifications

• Extremely strong software engineering skills.
• Experience designing and conducting data collection tasks, including working with human annotators.
• Strong statistical skills and experience evaluating scientific experiments related to data collection and model performance.
• Experience analysing datasets with respect to their quality, biases, and suitability for training ML models.
• Hands-on experience training large language models (LLMs) on distributed training infrastructures.
• Familiarity with evaluating and improving the generalisability and robustness of ML systems.
• Proficiency in programming languages such as Python and ML frameworks (e.g., PyTorch, TensorFlow, JAX).
• Excellent communication skills to collaborate effectively with cross-functional teams and present findings.
• One or more papers at top-tier venues (such as NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, EMNLP).

Responsibilities

• Design and oversee data collection tasks, including supporting human annotators and ensuring data quality.
• Develop and apply statistical methods to evaluate the quality and reliability of datasets.
• Analyse and assess the generalisability and robustness of ML systems across diverse use cases.
• Collaborate with teams to improve dataset quality and model performance.
• Train and fine-tune large language models (LLMs) on distributed training infrastructures.
• Conduct experiments to evaluate model performance and identify areas for improvement.

Skills

Software engineering

Data collection design

Statistical methods

Python

PyTorch

TensorFlow

JAX

Distributed training

LLMs

Communication

Research papers

Tools

PyTorch

TensorFlow

JAX

Job description

Cohere, a leading security-first enterprise AI company, is hiring a Member of Technical Staff in Data Analysis and Evaluation. You will design data-collection tasks, apply statistical methods, and assess model robustness.

Collaborate with researchers and engineers to improve dataset quality and model performance, including distributed training of LLMs. The role emphasizes rigorous experimentation, cross-functional teamwork, and contributing to high-quality AI systems that scale across diverse