Machine Learning Intern/Co-op (Winter 2024)

Key Skills

Analytical and Problem solving, CUDA/GPU programming, Effective communication skills, JAX framework, Machine learning techniques, Natural Language Processing (NLP), NeurIPS, Python Programming, Transformer

Job Description

Who are we?

We’re a small, diverse team working at the cutting edge of machine learning. At Cohere, our mission is to build machines that understand the world and to make them safely accessible to all. Language is at the crux of this, but it can be difficult and expensive to parse the syntax, semantics, and context that all work together to give words meaning. The Cohere platform provides access to Large Language Models through its APIs that read billions of web pages and learns to understand the meaning, sentiment, and intent of the words we use in a richness never seen before.

We've recently raised our Series C, and we are focused on bringing our technology to market. We will partner with customers so they can build natural language understanding and generation into their products with just a few lines of code.

We are focused on creating a diverse and inclusive work environment so that all of our team members can thrive. We welcome kind and brilliant people to our team, from wherever they come.

Why this role?

Ship state of the art models to production.

Design and implement novel research ideas.

Build elegant training/deployment pipelines.

Join us at a pivotal moment, shape what we build and wear multiple hats as an intern!

Please Note: To be eligible for this position you should be a student currently enrolled in a post-secondary program, available for a full-time 3-6 month internship, co-op, or research work term (January - April). We have offices in Toronto, Palo Alto, and London but embrace being remote-first! There are no restrictions on where you can be located for this role.

As a Machine Learning Intern, you will:

• Design, train and improve upon cutting-edge models
• Help us develop new techniques to train and serve models safer, better, and faster
• Train extremely large-scale models on massive datasets
• Explore continual and active learning strategies for streaming data
• Publish your work in top-tier conferences and journals
• Learn from experienced senior machine learning technical staff
• Work closely with product teams to develop solutions

You may be a good fit if you have:

• Proficiency in Python and related ML frameworks such as Tensorflow, TF-Serving, JAX, and XLA/MLIR
• Experience using large-scale distributed training strategies
• Familiarity with autoregressive sequence models, such as Transformers
• Strong communication and problem-solving skills
• A demonstrated passion for applied NLP models and products
• Bonus: experience writing kernels for GPUs using CUDA
• Bonus: experience training on TPUs
• Bonus: papers at top-tier venues (such as NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, EMNLP)

If some of the above doesn’t line up perfectly with your experience, we still encourage you to apply! If you consider yourself a thoughtful worker, a lifelong learner, and a kind and playful team member, Cohere is the place for you.

We value and celebrate diversity and strive to create an inclusive work environment for all. We welcome applicants of all kinds and are committed to providing an equal opportunity process. Cohere provides accessibility accommodations during the recruitment process. Should you require any accommodation, please let us know and we will work with you to meet your needs.