Staff Engineer, Safety for AI Agents

Cohere; Montreal (administrative region); Hybrid; CAD 120,000 - 180,000; Full time

Benefits offered by this job

Open and inclusive culture

Collaborative AI research environment

Weekly lunch stipend

Full health and dental benefits

100% Parental Leave top‑up

Personal enrichment benefits

Remote-flexible work options

6 weeks of vacation

Job summary

A leading AI research firm in Montreal seeks a Member of Technical Staff for its Safety for Agents team. This role focuses on enhancing the safety of Large Language Models by working on data generation and evaluation methods. Ideal candidates will have strong statistical and programming skills, as well as a passion for ethical AI development. Cohere offers a supportive work environment with excellent benefits, including a flexible work structure, health care, and generous leave policies.

Qualifications

• Strong statistical skills and experience evaluating scientific experiments.
• Extremely strong software engineering skills.
• Experience analyzing datasets with respect to their quality and biases.
• Hands-on experience training large language models on distributed infrastructures.
• Excellent communication skills to collaborate effectively.

Responsibilities

• Focus on data generation, post-training algorithms, and evaluation methods.
• Work closely with cross-functional teams.
• Handle messy data and scientific problems.

Skills

Statistical skills

Software engineering

Data collection design

Dataset analysis

Training large language models

Evaluating ML system robustness

Programming with Python

Communication skills

Tools

PyTorch

TensorFlow

JAX

Job description

A leading AI research firm in Montreal seeks a Member of Technical Staff for its Safety for Agents team. This role focuses on enhancing the safety of Large Language Models by working on data generation and evaluation methods. Ideal candidates will have strong statistical and programming skills, as well as a passion for ethical AI development. Cohere offers a supportive work environment with excellent benefits, including a flexible work structure, health care, and generous leave policies.