Waabi ML Training Optimization Engineer Role

Waabi • toronto, on, Canada • Posted June 01, 2026

Location toronto, on
Job Type Full-time
Category Other-General
Posted June 01, 2026
Empower the future of transportation as Waabi's ML Training Optimization Engineer. Focus on creating high-performance distributed training frameworks in a collaborative environment.
You will contribute to pioneering work in physical AI, developing efficient training processes for self-driving technologies. This role requires advanced technical skills to profile and optimize model performance, while collaborating with a team dedicated to driving innovation. Your work will ensure effective resource usage and facilitate broad adoption of optimal practices.
Key Responsibilities:
• Standardize and improve distributed training frameworks
• Profile model performance to identify bottlenecks
• Research and implement cutting-edge training technologies
• Partner with ML engineers for resource optimization
• Enhance tools for wider internal adoption
Requirements:
• MS/PhD or Bachelors with a minimum of 4 years experience
• Strong coding skills in Python, C++, or Rust<...

Interested in this role?

Click the button below to start your application.

Apply Now