[Remote] Member of Technical Staff, Modeling

Note: The job is a remote job and is open to candidates in USA. Cohere is a company focused on scaling intelligence to serve humanity through AI systems. As a Member of Technical Staff, you will design and implement novel research ideas, ship state-of-the-art models to production, and maintain deep connections to academia.

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

• 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
• 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!)

Company Overview