Member of Technical Staff, Training Infra Engineer

Highlights

• Ship state-of-the-art frontier models to production for enterprise customers.
• Improve training efficacy with tools that speed up large-scale training cycles.
• Join a team with one of the highest compute-to-engineer ratios worldwide.

What they offer

• Equity
• Weekly lunch stipend
• Full health and dental benefits including mental health budget
• RRSP matching, 401K and Pension Scheme
• 100% parental leave top-up up to 6 months
• 6 weeks paid vacation
• $500 home office stipend and co-working benefit
• Annual enrichment and education stipend

Ideal candidate

A strong software engineer passionate about large-scale model training who thrives bridging research and production. You love building performant infrastructure, learning from top researchers, and maximizing impact with abundant compute.

Requirements

• Extremely strong software engineering skills
• Proficiency in Python and related ML frameworks such as JAX, PyTorch and XLA/MLIR
• Experience with distributed training infrastructures (Kubernetes, Slurm) and associated frameworks (Ray)
• Experience using large-scale distributed training strategies
• Hands-on experience training large models at scale and contributing to training infrastructure tooling