Waabi

ML Training Optimization Senior Engineer

📍 Location
toronto, on
⏰ Job Type
Full-time
📅 Posted
June 09, 2026
Apply Now

Job Description

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 new 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 field with 4+ years of experience • Strong coding skills in Pyth...

Ready to Apply?

Take the next step in your career - we're hiring now!

Apply for this Position