Machine Learning Intern/Co-op
Are you applying to the internship?
Job Description
About Company:
Cohere is a leading AI research company with a mission to scale intelligence to serve humanity. They are focused on training and deploying frontier models for developers and enterprises building AI systems, particularly in areas like semantic search, RAG, and agents. They are a team of researchers, engineers, designers, and other professionals passionate about their craft and committed to building great products. They value diversity and strive for an inclusive work environment.
Job Description:
Cohere is seeking a Machine Learning Intern/Co-op to join their team. This is a full-time, 3-6 month internship opportunity for students currently enrolled in a post-secondary program.
As a Machine Learning Intern, you will:
• Design, train, and improve upon cutting-edge models.
• Help 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.
• Work closely with product teams to develop solutions.
Required Skills:
• 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 Skills:
• 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).
Benefits and Perks:
• An open and inclusive culture and work environment
• Work closely with a team on the cutting edge of AI research
• Free daily lunch (for interns, likely in office).
• 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, New York, San Francisco and London and coworking stipends
• Paid vacation
