Machine Learning Intern/Co-op
Are you applying to the internship?
Job Description
About Cohere:
Cohere is a leading AI research company with a mission to scale intelligence to serve humanity. They are training and deploying frontier models for developers and enterprises, focusing on building AI systems for semantic search, RAG, and agents. Cohere values innovation, hard work, and customer focus, and fosters a diverse and inclusive work environment.
Job Description: Machine Learning Intern/Co-op
This is a full-time internship opportunity for a Machine Learning Intern/Co-op at Cohere. The intern will be responsible for designing, training, and improving cutting-edge models, developing new techniques, and working closely with product teams.
Responsibilities:
• 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.
Requirements:
• Currently enrolled in a post-secondary program.
• Available for a full-time 3-6 month internship, co-op, or research work term.
• 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.
• Experience training on TPUs.
• Papers at top-tier venues (such as NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, EMNLP).
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, New York, San Francisco and London and coworking stipends.
• Paid vacation (Full-Time Employees Only: 6 weeks).
• Weekly Lunch Stipend, in-office lunches & snacks (Full-Time Employees Only).
• 100% Parental Leave top-up for 6 months for employees based in Canada, the US, and the UK (Full-Time Employees Only).
• Workspace improvement (Full-Time Employees Only).
