ML Training Optimization Senior Engineer (Toronto)

Join Waabi as a Senior Engineer focused on ML Training Optimization. Drive innovations in distributed training while optimizing performance metrics significantly. This role demands an MS/PhD or Bachelor’s degree alongside four years of industry experience in relevant areas like Robotics.

Your expertise in coding languages, especially Python and C++, along with familiarity with tools such as PyTorch, will be crucial. We are looking for a team-oriented individual who is passionate about self-driving solutions.

Key Responsibilities

• Build and standardize distributed training frameworks for stability
• Analyze model runtime and memory consumption for improvements
• Explore and integrate recent technologies in training setups
• Collaborate across teams for best practices on resource management
• Innovate tooling and dashboards for greater adoption

Requirements

• Degree in Computer Science or relevant fiel...