Staff Engineer – Training Infra & ML Systems

Cohere; Greater London; Remote; GBP 70,000 - 90,000; Full time

Benefits offered by this job

Open and inclusive culture

Weekly lunch stipend of $75/£75

Full health and dental benefits

6 weeks of paid vacation (30 working days)

Remote-flexible working options

Job summary

Cohere is seeking a Member of Technical Staff to aid in AI model training and deployment. The role involves designing high-performance software and improving the training infrastructure, alongside a team of leading researchers and engineers. Cohere embraces a remote-friendly policy, with offices in London and other major cities.

We offer a range of benefits including a lunch stipend, health benefits, generous vacation, and a commitment to diversity and inclusion.

Qualifications

• Experience with associated frameworks for distributed training infrastructures.
• Contributed to the tooling and/or setup of training infrastructure.
• Bonus: publication at top-tier AI venues.

Responsibilities

• Design and implement scalable software for training AI models.
• Improve system performance for training setup.
• Craft tools to enhance training efficacy.
• Conduct research on AI infrastructure.

Skills

Extremely strong software engineering skills

Proficiency in Python

Experience with distributed training infrastructures

Hands-on experience training large models

Tools

JAX

PyTorch

Kubernetes

Ray

Job description

Cohere is seeking a Member of Technical Staff to aid in AI model training and deployment. The role involves designing high-performance software and improving the training infrastructure, alongside a team of leading researchers and engineers. Cohere embraces a remote-friendly policy, with offices in London and other major cities.

We offer a range of benefits including a lunch stipend, health benefits, generous vacation, and a commitment to diversity and inclusion.