Remote ML Intern: Build Cutting-Edge Models & Pipelines

Cohere; United States; Hybrid; USD 25,000 - 35,000; Full time

Benefits offered by this job

Weekly lunch stipend

Health and dental benefits

RRSP matching/401K

Parental leave + enrichment

6 weeks paid vacation

Office stipend for remote work

Job summary

Cohere is seeking a Machine Learning Intern to design and train frontier models, build training and deployment pipelines, and work with product teams. You will learn from senior ML staff, explore new techniques for safe and efficient training, and contribute to large-scale projects on time-limited terms.

To be eligible, you must be enrolled in a post-secondary program and available for a full-time 3–6 month internship, co-op, or research work term. We value curiosity and collaboration.

Qualifications

• Proficiency in Python and ML frameworks (TensorFlow, TF-Serving, JAX, XLA/MLIR).
• Experience with large-scale distributed training strategies.
• Familiarity with autoregressive sequence models (Transformers).
• Strong communication and problem-solving skills.
• Passion for applied NLP models and products.
• Bonus: CUDA kernel experience; TPU training experience; relevant papers.

Responsibilities

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

Skills

Python

ML frameworks (Tensorflow, TF-Serving,

JAX

XLA/MLIR

distributed training

Transformers

NLP enthusiasm

communication

Education

Post-secondary student (internship)

Tools

CUDA

TPU

Job description

Cohere is seeking a Machine Learning Intern to design and train frontier models, build training and deployment pipelines, and work with product teams. You will learn from senior ML staff, explore new techniques for safe and efficient training, and contribute to large-scale projects on time-limited terms.

To be eligible, you must be enrolled in a post-secondary program and available for a full-time 3–6 month internship, co-op, or research work term. We value curiosity and collaboration.