ML Research Intern: Build & Deploy Frontier Models

SpeedyApply LLC; San Francisco (CA); Hybrid; USD 40,000 - 60,000; Full time

Benefits offered by this job

Weekly lunch stipend

Full health benefits

Family/Parental leave

Conference & learning stipend

Home office stipend

Job summary

Cohere is a leading security-first enterprise AI company building foundation models and end-to-end products to solve real-world business problems. We’re hiring a Machine Learning Intern to ship state-of-the-art models to production and design novel training/deployment pipelines.

As an intern, you will design and train models, explore new learning strategies for streaming data, and collaborate with product teams to deliver practical NLP solutions.

Qualifications

• 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.
• Bonus: experience training on TPUs.
• Bonus: papers at top-tier venues (NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, EMNLP).

Responsibilities

• Design, train and improve upon cutting-edge models.
• Help us 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.

Skills

Python

TensorFlow

JAX

XLA/MLIR

Distributed training

Transformers

Communication

NLP interest

CUDA

TPU

Tools

TF-Serving

Job description

Cohere is a leading security-first enterprise AI company building foundation models and end-to-end products to solve real-world business problems. We’re hiring a Machine Learning Intern to ship state-of-the-art models to production and design novel training/deployment pipelines.

As an intern, you will design and train models, explore new learning strategies for streaming data, and collaborate with product teams to deliver practical NLP solutions.