Machine Learning Intern/Co-op (Fall, 2026)
Core
Design, train, and improve cutting-edge large-scale AI models (NLP, Transformers) for developers and enterprises building AI systems.
Role type
Machine Learning Intern (Research & Engineering)
Builds
Production-ready state-of-the-art models and training/deployment pipelines
Domain
Artificial Intelligence / Natural Language Processing / Large Language Models
Deliverable
production ML models
Required skills
Python, TensorFlow, JAX, XLA/MLIR, distributed training strategies, autoregressive sequence models (Transformers)
Preferred skills
CUDA kernel development, TPU training experience, publications at top-tier venues (NeurIPS, ICML, ACL, etc.)
Technologies
TensorFlow, TF-Serving, JAX, XLA/MLIR, CUDA, TPUs
Responsibilities
Design and implement novel research ideas; Train extremely large-scale models on massive datasets; Develop new techniques to train and serve models safer, better, and faster; Work closely with product teams to develop solutions
Seniority
Intern (Student)