Machine Learning Intern/Co-op (Winter 2027)

Job summary

Responsibilities

We’re training and deploying frontier models for enterprises who are building AI systems. We believe that our work is instrumental to the widespread adoption of AI and we are looking for folks that want to be part of that.

• 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.

What they're looking for

• 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.

Tech stack

• Python
• TensorFlow
• JAX
• Transformers
• CUDA