Staff Engineer, Safety & Security for AI Agents

Jaide Health; Toronto; Remote; CAD 180,000 - 240,000; Full time

Benefits offered by this job

Weekly lunch stipend

Health and dental benefits

RRSP matching

401K

Pension scheme

Parental leave top-up

Enrichment benefits

Education stipend

6 weeks paid vacation

Travel budget

Annual company offsite

Home office stipend

Job summary

Cohere is seeking a Member of Technical Staff in the Safety for Agents team to advance trustworthy, safe, and scalable LLMs. You will work on data generation, post-training algorithms, and evaluation methods across cross-functional teams.

This remote-friendly role emphasizes collaboration with product and policy teams, distributed training, and rigorous experimentation. You will join a global company with offices in Toronto, London, New York, and more.

Qualifications

• Strong statistical skills and experience evaluating scientific experiments related to data collection and model performance.
• Extremely strong software engineering skills.
• Strong expertise in designing and conducting data collection tasks, including working with human annotators.
• Experience analyzing 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 generalizability 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

• Strong statistical skills and experience evaluating experiments related to data collection and model performance.
• Extremely strong software engineering skills.
• Designing and conducting data collection tasks, including working with human annotators.
• Analyze datasets for quality, biases, and suitability for training ML models.
• Hands-on experience training large language models (LLMs) on distributed training infrastructures.
• Evaluate generalizability and robustness of ML systems.
• Proficiency in Python and ML frameworks (PyTorch, TensorFlow, JAX).
• Excellent communication with cross-functional teams and presenting findings.
• Publish papers at top-tier venues (NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, EMNLP).

Skills

Statistical skills

Software engineering

Data collection

Dataset analysis

LLM training

Model evaluation

Python

ML frameworks

Communication

Research papers

Job description

Cohere is seeking a Member of Technical Staff in the Safety for Agents team to advance trustworthy, safe, and scalable LLMs. You will work on data generation, post-training algorithms, and evaluation methods across cross-functional teams.

This remote-friendly role emphasizes collaboration with product and policy teams, distributed training, and rigorous experimentation. You will join a global company with offices in Toronto, London, New York, and more.