Member of Technical Staff, Modeling

Cohere

full-time

Posted on: 3/18/2026

Location Type: Remote

✨ AI Apply

About the role

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.

Extremely strong software engineering skills.
Proficiency in Python and related ML frameworks such as Tensorflow, TF-Serving, JAX, and 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: paper at top-tier venues (such as NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, EMNLP).

Benefits

An open and inclusive culture and work environment
Work closely with a team on the cutting edge of AI research
Weekly lunch stipend, in-office lunches & snacks
Full health and dental benefits, including a separate budget to take care of your mental health
100% Parental Leave top-up for up to 6 months
Personal enrichment benefits towards arts and culture, fitness and well-being, quality time, and workspace improvement
Remote-flexible, offices in Toronto, New York, San Francisco, London and Paris, as well as a co-working stipend
6 weeks of vacation (30 working days!)

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools

software engineeringPythonTensorflowTF-ServingJAXXLAMLIRCUDAdistributed trainingautoregressive sequence models