Machine Learning Intern/Co-op (Fall, 2026)

📋 Description

• Design, train and improve upon cutting-edge models.
• Build elegant training/deployment pipelines.
• Design and implement novel research ideas.
• Train extremely large-scale models on massive datasets.
• Explore continual and active learning strategies for streaming data.
• Work closely with product teams to develop solutions.

🎯 Requirements

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

🎁 Benefits

• An open and inclusive culture and work environment
• Work with a team on the cutting edge of AI research
• Weekly lunch stipend, in-office lunches & snacks
• Full health and dental benefits, including mental health budget
• 100% Parental Leave top-up for up to 6 months
• Remote-flexible, offices in major cities with a co-working stipend