Staff Research Engineer, RL & Model Integration (Remote-friendly)

Cohere; Montreal; Hybrid; CAD 80,000 - 120,000; Full time

Benefits offered by this job

Open and inclusive work environment

Weekly lunch stipend

Full health and dental benefits

Parental leave top-up

Personal enrichment benefits

Remote-flexible work

6 weeks of vacation

Job summary

A leading AI research company in Montreal seeks a Member of Technical Staff to develop and scale machine learning algorithms. The role requires strong software engineering skills and proficiency in Python and ML frameworks. The ideal candidate will collaborate with engineering teams and contributes to innovative LLM training methods. The position offers competitive benefits, including extensive parental leave and a remote-friendly work culture.

Qualifications

• Extremely strong software engineering skills.
• Value test-driven development, clean code, and reducing technical debt.
• Proficiency in Python and ML frameworks.

Responsibilities

• Design and write high-performing and scalable software for training models.
• Develop new tools to support and accelerate research and LLM training.
• Coordinate with other engineering and scientific teams.

Skills

Software engineering skills

Test-driven development

Proficiency in Python

Experience with ML frameworks

Distributed training strategies

Tools

JAX

PyTorch

Kubernetes

Ray

Job description

A leading AI research company in Montreal seeks a Member of Technical Staff to develop and scale machine learning algorithms. The role requires strong software engineering skills and proficiency in Python and ML frameworks. The ideal candidate will collaborate with engineering teams and contributes to innovative LLM training methods. The position offers competitive benefits, including extensive parental leave and a remote-friendly work culture.