Waabi

Senior Engineer in ML Training Optimization

📍 Location
toronto, on
⏰ Job Type
Full-time
📅 Posted
June 04, 2026

Job Description

Elevate your career as a Senior ML Training Optimization Engineer at Waabi. Focus on building advanced distributed training frameworks while enhancing performance and efficiency.

In this senior position, you will possess an MS/PhD or a Bachelor’s degree paired with a minimum of four years in Computer Science or related fields. You must demonstrate proficiency in multiple programming languages like Python and C++, as well as hands-on experience with deep learning tools like PyTorch. Solution-oriented and collaborative, you'll thrive in a role that pushes the boundaries of self-driving tech.

Key Responsibilities: • Develop robust distributed training frameworks for both research and production • Profile model runtime and memory to eliminate bottlenecks • Investigate and implement emerging technologies into training frameworks • Collaborate with teams for optimal resource utilization • Design and enhance tools to promote your work effectively

Requirements: • MS/...

Ready to Apply?

Take the next step in your career journey with Waabi

Apply Now