5 months ago
Toronto, Canada +3 moreIntern
Responsibilities
- Design, train, and improve cutting-edge models.
- Develop techniques to train and serve models more safely, effectively, and efficiently.
- Train extremely large-scale models on massive datasets.
- Explore continual and active learning strategies for streaming data.
- Design and implement novel research ideas.
- Build training and deployment pipelines.
- Work closely with product teams to develop solutions.
Requirements
- Currently enrolled in a post-secondary program and available for a full-time 3–6 month internship, co-op, or research work term.
- Proficiency in Python and related machine learning 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.
- Demonstrated passion for applied natural language processing models and products.
- Strong communication and problem-solving skills.
- Experience writing GPU kernels using CUDA is a bonus.
- Experience training on TPUs is a bonus.
- Papers at venues such as NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, or EMNLP are a bonus.
Benefits
- Full-time employees receive a weekly lunch stipend, in-office lunches and snacks, health and dental benefits, mental health support, parental leave top-up, personal enrichment benefits, remote-flexible work, coworking stipend, and six weeks of vacation.
- The role is a full-time 3–6 month internship, co-op, or research work term.
- Offices are located in Toronto, New York, San Francisco, London, and Paris.
Tech Stack
PythonTensorFlow
Categories
AI ResearchML Engineering
About Cohere
Cohere builds large language models and an enterprise AI platform that companies use for search, summarization, and workflow automation, delivered via API or private deployments. Founded in 2019 and headquartered in Toronto, it focuses on multilingual models, data controls, and options to run across major clouds or on-premises. The business is privately held and serves security- and compliance-sensitive organizations.