AI ML Optimization Engineer at Waabi

Waabi

vancouver, metro vancouver regional district, Canada
Full-time
Posted June 04, 2026

Job Description

Elevate self-driving technology with Waabi as a Senior ML Optimization Engineer. Collaborate on AI-driven solutions and optimize training frameworks for enhanced stability and efficiency.
Waabi is seeking a Senior ML Optimization Engineer to develop next-gen self-driving systems. You will work closely with autonomy and algorithm engineers to implement an AI-first approach in our training infrastructure while tackling performance bottlenecks. Ideal candidates will have a strong foundation in deep learning frameworks, coding skills in various languages, and experience with distributed training.
Key Responsibilities:
• Collaborate on building self-driving systems with engineers
• Develop stable and efficient distributed training frameworks
• Profile model runtime and memory for performance issues
• Evaluate new technologies for training and inference
• Integrate efficient CUDA kernels for enhanced processes
Requirements:
• MS/PhD or Bachelor's degree with 6+ yea...