Staff AI Modeling Engineer

Cohere; New York (NY); Remote; USD 180,000 - 240,000; Full time

Benefits offered by this job

A weekly lunch stipend for lunch

Full health and dental benefits

RRSP matching / 401K / Pension

Parental Leave top-up

Education & learning stipend

Paid vacation

Home-office stipend

Travel budget to other offices

Job summary

Cohere is seeking a talented AI systems researcher/engineer to design and implement novel research ideas, ship state-of-the-art models to production, and collaborate with world-class researchers. The role blends engineering and research across large-scale compute and data infrastructure, with remote-friendly options and Global offices.

You will work on building AI systems, exploring new architectures, and contributing to production code and research, with strong Python and ML framework skills.

Qualifications

• Extremely strong software engineering skills.
• Proficiency in Python and TensorFlow, JAX, XLA/MLIR.
• Experience writing kernels for GPUs using CUDA.
• Experience using large-scale distributed training strategies.
• Familiarity with autoregressive sequence models, such as Transformers.
• Bonus: papers at top-tier venues.

Responsibilities

• Design, build and scale AI systems for serving our users.
• Research, implement, and experiment with ideas on our supercompute and data infrastructure.
• Learn from and work with the best researchers in the field.

Skills

Python

TensorFlow

JAX

CUDA

Distributed training

Transformers

Research collaboration

Tools

TF-Serving

XLA/MLIR

CUDA Kernels

Job description

Cohere is seeking a talented AI systems researcher/engineer to design and implement novel research ideas, ship state-of-the-art models to production, and collaborate with world-class researchers. The role blends engineering and research across large-scale compute and data infrastructure, with remote-friendly options and Global offices.

You will work on building AI systems, exploring new architectures, and contributing to production code and research, with strong Python and ML framework skills.