over 1 year ago
Toronto, Canada +5 moreMid Level / Senior
H1B Sponsor
Responsibilities
- Design, build, and scale AI systems for user service.
- Research, implement, and experiment with ideas on supercompute and data infrastructure.
- Collaborate with leading researchers in the AI field.
Requirements
- Extremely strong software engineering skills.
- Proficiency in Python and related ML frameworks such as Tensorflow, TF-Serving, JAX, and XLA/MLIR.
- Experience writing kernels for GPUs using CUDA.
- Experience with large-scale distributed training strategies.
- Familiarity with autoregressive sequence models, such as Transformers.
- Bonus: published papers at top-tier venues in AI.
Benefits
- An open and inclusive culture and work environment.
- Weekly lunch stipend, in-office lunches, and snacks.
- Full health and dental benefits, including a budget for mental health.
- 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).