Applied ML Staff Engineer - Enterprise AI

Visa Hunt; Greater London; Hybrid; GBP 120,000 - 170,000; Full time

Benefits offered by this job

Lunch stipend

Health and dental benefits

RRSP matching, 401K, Pension Scheme

Parental Leave top-up

Education & learning stipend

6 weeks of paid vacation

Travel to other offices / annual off (

Home office stipend

Job summary

Cohere in London is seeking a Member of Technical Staff for the Applied ML team to design and deploy large language model solutions in collaboration with customers.

You will mentor engineers, set technical direction, and work across product, legal, and partner teams to deliver impactful ML products at scale.

This role offers a remote-friendly culture with offices in major cities and generous benefits including lunch stipend, health/dental, and learning stipends.

Qualifications

• Experience leading ML teams in applied research and production.
• Experience setting and executing technical direction for product development.
• Experience working cross-discipline in matrixed organizations.
• Proficiency in Python and ML frameworks such as TensorFlow, TF-Serving, JAX, and XLA/MLIR.
• Experience using large-scale distributed training and inference.
• Strong mentorship, communication, and problem-solving skills.
• A demonstrated passion for applied ML models and products.

Responsibilities

• Plan and execute large-group projects that carry through from ideation to production.
• Bring cross-functional alignment across engineering, product and other disciplines.
• Mentor a distributed team of engineers in subject matter expertise.
• Identify opportunities and gaps in existing models and strategize what to work on.
• Work closely with product teams to develop solutions.
• Engage in collaborations with our partner organizations.
• Assist our legal teams with preparation of patents on developed IP.
• Join us at a pivotal moment, shape what we build and wear multiple hats!

Skills

Leading ML teams

Technical direction

Cross-functional collaboration

Python

TensorFlow

TF-Serving

JAX

XLA/MLIR

Distributed training

Mentorship

Communication

Applied ML

B2B ML experience

Tools

TensorFlow

TF-Serving

JAX

XLA/MLIR

Job description

Cohere in London is seeking a Member of Technical Staff for the Applied ML team to design and deploy large language model solutions in collaboration with customers.

You will mentor engineers, set technical direction, and work across product, legal, and partner teams to deliver impactful ML products at scale.

This role offers a remote-friendly culture with offices in major cities and generous benefits including lunch stipend, health/dental, and learning stipends.