Senior / Staff ML Onboard Optimization Engineer (Québec City)

Responsibilities

• Collaborate closely with autonomy and algorithm engineers to scale safe self-driving systems using an AI‑first approach.
• Build standardized distributed training frameworks for research and production, driving our training towards current levels of stability and efficiency.
• Comprehensively profile model runtime and memory to pinpoint performance bottlenecks.
• Identify and evaluate emerging technologies that can be adopted into Waabi’s training and inference frameworks, including but not limited to efficient CUDA kernels for training, quantization, model exporting and compilation for inference, etc.

Qualifications

• MS/PhD or bachelor’s degree with a minimum of 6 years of industry experience in Computer Science, Robotics and/or similar technical field(s) of study.
• Solid coding proficiency in a variety of coding languages including Python, C++ or Rust.
• Experience in deep learning fra...