ML Optimization Engineer at Waabi

Waabi

Full-time Engineering
Apply Now
Location
quebec, capitale nationale, Canada
Posted
June 06, 2026

Job Description

Drive the future of self-driving tech with Waabi as a Senior Machine Learning Optimization Engineer. Utilize your expertise to enhance AI systems, focusing on stability and efficiency.

At Waabi, we are seeking an experienced ML Optimization Engineer with a minimum of six years in Computer Science or Robotics. Collaborate with autonomy teams, build distributed training frameworks, and improve model runtime profiling. Your skills in coding languages like Python or C++, along with deep learning frameworks, will be essential for our projects.

Key Responsibilities:
β€’ Scale safe self-driving systems using an AI-first methodology
β€’ Build standardized frameworks for distributed training
β€’ Profile model runtime and memory to identify bottlenecks
β€’ Evaluate emerging technologies for training and inference
β€’ Collaborate with algorithm engineers to enhance model performance

Requirements:
β€’ MS/PhD or Bachelor’s with 6+ years...