Staff Engineer, Safety for AI Agents

Cohere; New York (NY); Hybrid; USD 120,000 - 150,000; Full time

Benefits offered by this job

Inclusive work culture

Work with cutting-edge AI technology

Weekly lunch stipend and snacks

Full health and dental benefits

Parental leave top-up for 6 months

Personal enrichment benefits

Remote flexibility and co-working stipend

6 weeks of vacation

Job summary

A leading AI technology company in New York seeks a Member of Technical Staff focused on developing safer and more trustworthy Large Language Models (LLMs). The role requires strong statistical and software engineering skills, with responsibilities including data generation and collaboration with cross-functional teams. Candidates should have hands-on experience training LLMs and be proficient in Python and related ML frameworks. The company values diversity and offers a flexible work environment with competitive benefits.

Qualifications

• Proven statistical skills in evaluating scientific experiments.
• Strong software engineering capabilities.
• Experience managing data collection with human annotators.
• Skilled in dataset quality evaluation for ML training.
• Hands-on training of large language models on distributed systems.
• Expertise in enhancing ML system robustness.

Responsibilities

• Develop better and safer Large Language Models (LLMs).
• Collaborate with machine learning and data annotation teams.
• Tackle new scientific problems in model safety.

Skills

Strong statistical skills

Strong software engineering skills

Experience with data collection tasks

Experience analyzing datasets

Hands-on experience training large language models

Familiarity with ML systems

Proficiency in Python

Excellent communication skills

Publications in top-tier venues

Tools

Python

PyTorch

TensorFlow

JAX

Job description

A leading AI technology company in New York seeks a Member of Technical Staff focused on developing safer and more trustworthy Large Language Models (LLMs). The role requires strong statistical and software engineering skills, with responsibilities including data generation and collaboration with cross-functional teams. Candidates should have hands-on experience training LLMs and be proficient in Python and related ML frameworks. The company values diversity and offers a flexible work environment with competitive benefits.