ML Intern/Co-op — Build & Ship Frontier Models (Remote)

Cohere; United States; Remote; USD 28,000 - 39,000; Full time

Benefits offered by this job

Lunch stipend

Health benefits

RRSP/401K

Parental leave

Paid vacation

Travel budget

Home office stipend

Job summary

Cohere is an AI company focused on security-first enterprise models and end-to-end products. This internship offers a chance to work on state-of-the-art ML models and training pipelines, contributing to production-ready systems.

You will design, train, and refine models, explore safety and efficiency techniques, and collaborate with product teams to deploy impactful NLP solutions. This internship requires being enrolled in a post-secondary program and committing 3–6 months full-time.

Qualifications

• Proficiency in Python and ML frameworks like TensorFlow, TF-Serving, JAX, and XLA/MLIR.
• Experience with large-scale distributed training strategies.
• Familiarity with autoregressive sequence models such as Transformers.
• Strong communication and problem-solving skills.

Responsibilities

• Design, train and improve upon cutting-edge models.
• Develop new techniques to train and serve models safely, better, and faster.
• Train very large-scale models on massive datasets.
• Explore continual and active learning strategies for streaming data.
• Learn from senior ML staff.
• Collaborate with product teams to develop solutions.

Skills

Python

TensorFlow

JAX

XLA/MLIR

Transformers

Distributed training

NLP

Strong communication

Education

Current student in post-secondary program

Tools

TensorFlow Serving

CUDA

TPUs

Job description

Cohere is an AI company focused on security-first enterprise models and end-to-end products. This internship offers a chance to work on state-of-the-art ML models and training pipelines, contributing to production-ready systems.

You will design, train, and refine models, explore safety and efficiency techniques, and collaborate with product teams to deploy impactful NLP solutions. This internship requires being enrolled in a post-secondary program and committing 3–6 months full-time.