Machine Learning Intern/Co-op (Spring/Summer 2026)

About the Role

The Machine Learning Intern role is a unique opportunity to gain hands-on experience in the field of AI research and development. You will be working closely with experienced senior machine learning technical staff, who will provide guidance and support as you learn and grow in your role. As a member of the Cohere team, you will be part of a dynamic and innovative work environment that values diversity, inclusivity, and creativity.

What You Will Do

• Design, train, and improve upon cutting-edge models using various machine learning frameworks and tools
• 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
• Learn from experienced senior machine learning technical staff and contribute to the development of best practices
• Work closely with product teams to develop solutions that meet the needs of Cohere's customers
• Participate in code reviews and contribute to the improvement of the codebase
• Stay up-to-date with the latest developments in the field of AI research and development
• Collaborate with other teams to identify and prioritize project requirements

What We Are Looking For

• 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
• Experience with version control systems such as Git
• Strong understanding of computer science fundamentals, including data structures and algorithms
• Ability to work in a fast-paced environment and adapt to changing priorities

Nice to Have

• Experience writing kernels for GPUs using CUDA
• Experience training on TPUs
• Papers at top-tier venues (such as NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, EMNLP)
• Experience with cloud-based machine learning platforms such as Google Cloud AI Platform or Amazon SageMaker
• Familiarity with agile development methodologies and continuous integration/continuous deployment (CI/CD) pipelines

Benefits and Perks

• Opportunity to work on state-of-the-art models and contribute to the development of cutting-edge AI technologies
• Collaborative and dynamic work environment that values diversity, inclusivity, and creativity
• Access to cutting-edge tools and technologies, including large-scale distributed training infrastructure
• Professional development opportunities, including training and support for ongoing learning and growth
• Competitive compensation and benefits package, including full health and dental benefits, parental leave top-up, and a separate budget for mental health support
• Flexible work arrangements, including remote work options and a co-working stipend
• Access to a network of experienced professionals in the field of AI research and development
• Opportunities for career advancement and professional growth within the company

• Develop a strong foundation in machine learning fundamentals, including supervised and unsupervised learning, deep learning, and neural networks.
• Build a portfolio of personal projects that demonstrate your skills and experience in machine learning, such as models you've trained and deployed.
• Stay up-to-date with the latest developments in the field of AI research and development by attending conferences, meetups, and online courses.
• Practice communicating complex technical ideas to non-technical audiences, as this is a key skill for success in this role.
• Be prepared to ask questions and seek feedback during the interview process, as this demonstrates your willingness to learn and grow.
• Research the company culture and values to ensure they align with your own, and be prepared to discuss how you can contribute to the team's mission and goals.