Research Internship (AI)

TL;DR

Research Internship (AI/LLM): Designing and implementing novel research ideas to ship state-of-the-art large language models with an accent on base model training, RAG, safety, and finetuning. Focus on expanding the frontier of knowledge in language modelling, optimization, and multimodal systems.

Location: Remote (Canada, US, UK, Europe). Must have valid work authorization in the country of employment.

Company

Cohere is an AI research company focused on training and deploying frontier models for developers and enterprises to power semantic search, RAG, and agents.

What you will do

• Conduct cutting-edge machine learning research to build and train large language models.
• Focus on research projects aimed at expanding the frontier of knowledge in language modelling, evaluation, and multimodal models.
• Disseminate research results through the production of publications, datasets, and code.
• Contribute to research initiatives that have practical applications in Cohere's product development.

Requirements

• Currently pursuing a PhD in Machine Learning, NLP, or a related discipline (exceptional non-PhD candidates may be considered).
• Available for a full-time internship lasting 4-6 months.
• Must be eligible for work authorization in the country of employment at the time of hire.
• Experience with large-scale distributed training strategies, data annotation, and evaluation pipelines.
• Familiarity with autoregressive sequence models, specifically Transformers.
• Proficiency in Python, C, C++, or Lua, and frameworks such as JAX, PyTorch, or TensorFlow.

Nice to have

• Demonstrated expertise through publications in top-tier ML/NLP venues.
• Proven ability to tackle analytical problems using quantitative methodologies.
• Experience applying theoretical and empirical research to real-world problem-solving.

Culture & Benefits

• Inclusive culture working with a team on the cutting edge of AI research.
• Remote-flexible policy with offices in major cities and a co-working stipend.
• Comprehensive health and dental benefits, including a dedicated mental health budget.
• Generous vacation policy with 6 weeks (30 working days) of leave.
• 100% Parental Leave top-up for up to 6 months.
• Weekly lunch stipend, in-office lunches, and snacks.