Member of Technical Staff, Modeling

Stack

Responsibilities

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

Requirements

• ▹Extremely strong software engineering skills
• ▹Proficiency in Python and related ML frameworks such as TensorFlow, TF-Serving, JAX and XLA/MLIR
• ▹Experience writing GPU kernels using CUDA
• ▹Experience using large-scale distributed training strategies
• ▹Familiarity with autoregressive sequence models such as Transformers

Nice to have

• ▹Paper at a top-tier venue (NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, EMNLP)

Soft skills

What we offer

• ▹Weekly lunch stipend of $75/£75 or local currency equivalent
• ▹Full health and dental benefits, plus a separate mental health budget
• ▹RRSP matching, 401K, pension scheme
• ▹100% parental leave top-up for up to 6 months, for either parent
• ▹Annual enrichment and learning stipends
• ▹6 weeks of paid vacation (30 working days)
• ▹Travel budget, annual company offsite, $500 home office stipend, remote-friendly culture

About the company

Cohere is the leading security-first enterprise AI company, building cutting-edge foundation models and end-to-end products that solve real-world business problems. Co-headquartered in Toronto and San Francisco, with offices in London, New York City, Montreal, Seoul, Germany and Paris.