Machine Learning Intern/Co-op

Key Skills

Algorithms, AWS, Cisco CCNA certification, Framework, Google Analytics, GPUs, JAX framework, Large Language Models - LLMs, Machine learning techniques, Natural Language Processing (NLP), Python Programming, TensorFlow

Job Description

Cohere is focused on building and deploying large language model (LLM) AI into enterprises in a safe and responsible way that drives human productivity, and creates magical new ways to interact with technology and real business value. We’re a team of highly motivated and experienced engineers, innovators, and disruptors looking to change the face of technology.

Our goals are ambitious, but also concrete and practical. Cohere wants to fundamentally change how businesses operate, making everyone more productive and able to focus on doing better what they do best. Every day, our team breaks new ground, as we build transformational AI technology and products for enterprise and developers to harness the power of LLMs.

Cohere was founded by three global leaders in AI development, including our CEO, Aidan Gomez, who co-created the Transformer, which makes LLMs possible. Collectively, we're driven by the belief that our technology has the potential to revolutionize the way enterprises, their employees, and customers engage with technology through language.

Cohere’s broader research team is world-renowned, having contributed to the development of sentence transformers for semantic search, dynamic adversarial data collection and red teaming, and retrieval augmented generation, often referred to as “RAG,” among other technological breakthroughs.

We have been deliberate in assembling a team of operational leaders with industry-leading experience, with backgrounds working at the most sophisticated, demanding, and respected enterprises in the world. Cohere’s operational leaders have built, scaled, and led multi-billion product lines and businesses at Google, Apple, Rakuten, YouTube, AWS, and Cisco.

The Cohere team is a collective from all walks of life, from people who left college to start businesses, to some of the most experienced people from globally renowned companies. We believe a diverse team is the key to a safer, more responsible technology, and that different experiences and backgrounds enable us to tackle problems from all angles and avoid blindspots.

There’s no better time to play a role in defining the future of AI, and its impact on the world.

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.

Our Perks:

• An open and inclusive culture and work environment
• Work closely with a team on the cutting edge of AI research
• Free daily lunch
• Full health and dental benefits, including a separate budget to take care of your mental health
• Personal enrichment benefits towards arts and culture, fitness and well-being
• Remote-flexible, offices in Toronto, Palo Alto, and London and coworking stipend
• Paid vacation