12 days ago
Toronto, Canada +3 moreIntern
H1B Sponsor
Responsibilities
- Design, train, and improve cutting-edge machine learning models.
- Develop new techniques for safer and faster model training and serving.
- Train large-scale models on massive datasets.
- Explore continual and active learning strategies for streaming data.
- Collaborate with product teams to develop effective solutions.
Requirements
- Proficiency in Python and ML frameworks like Tensorflow, JAX, and TF-Serving.
- 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 NLP models and products.
- Bonus: experience with CUDA for GPU kernels.
- Bonus: experience training on TPUs.
- Bonus: publications at top-tier venues in AI.
Benefits
- Open and inclusive culture and work environment.
- Work closely with a cutting-edge AI research team.
- Weekly lunch stipend, in-office lunches, and snacks.
- Full health and dental benefits, including mental health support.
- 100% parental leave top-up for up to 6 months.
- Personal enrichment benefits for arts, culture, fitness, and workspace improvement.
- Remote-flexible work options with offices in major cities.
- 6 weeks of vacation (30 working days).