Machine Learning Intern/Co-op
This internship focuses on designing, training, and deploying large-scale machine learning models, exploring learning strategies, and partnering with product teams. Candidates should be enrolled students with Python and ML framework proficiency, distributed training experience, and familiarity with Transformer-based models.
About the job
Responsibilities
- Design, train, and improve cutting-edge machine learning models.
- Develop techniques to train and serve models more safely, effectively, and efficiently.
- 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
- Currently enrolled in a post-secondary program.
- Available for a full-time 3–6 month internship, co-op, or research work term.
- Proficiency in Python and related machine learning frameworks, including TensorFlow, TF-Serving, JAX, and XLA/MLIR.
- Experience with large-scale distributed training strategies.
- Familiarity with autoregressive sequence models such as Transformers.
- Strong communication and problem-solving skills.
- Demonstrated passion for applied natural language processing models and products.
Nice-to-haves
- Experience writing GPU kernels using CUDA.
- Experience training on TPUs.
- Papers at top-tier venues such as NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, or EMNLP.
Compensation and Benefits
- Weekly lunch stipend of $75/£75 or equivalent in local currency.
- Full health and dental benefits, including a separate mental health budget.
- RRSP matching, 401(k), and pension scheme.
- Parental leave top-up for up to 6 months for either parent.
- Annual enrichment benefits for arts and culture, fitness and wellness, quality time, and workspace improvements.
- Education and learning stipend for conferences, courses, and coaching.
- Six weeks of paid vacation.
- Remote employees may receive a budget for travel to other offices and an annual company offsite.
- Coworking benefit and a $500 home office stipend for remote employees.
Skills
Python, TensorFlow, Tf-Serving, JAX, Xla, Mlir, Distributed Training, Transformers, Natural Language Processing, CUDA, Tpus
Similar jobs
ML Engineering jobs
DataVisor
Mountain View, CA
Builds high-scale data pipelines, distributed systems, and AI agent workflows using LLMs for fraud intelligence platform. Requires 2+ years software engineering, Python proficiency, big data tools, AWS/K8s, and ML foundations.
Coinbase
San Francisco, CA
Machine learning intern pursuing a Ph.D. who will build and deploy production-scale models and pipelines, conduct an end-to-end research project, and collaborate with engineering and product teams on blockchain and cryptocurrency applications.
Fireworks AI
San Mateo, CABuild, deploy, and optimize AI applications and machine learning models for customer use cases while contributing to an internal ML platform. This new graduate role requires a technical master’s degree, hands-on ML or LLM experience, and strong customer communication skills.
PathAI
Boston, MADevelop and deploy machine learning models for biomedical research and AI product development, collaborating with scientific, engineering, and product teams. Requires a master's degree with 2–4 years of experience or a PhD with 0–2 years, plus strong Python and ML development skills.
Earnin
Mountain View, CA
Machine learning engineer who trains, evaluates, and productionizes models and LLM-powered applications for financial products. Requires 2+ years of ML systems experience, strong Python and PyTorch skills, production data pipelines, model evaluation, and API development.