Senior / Staff ML Onboard Optimization Engineer (Moncton)

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, drive our training towards new 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 Bachelors 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 frameworks...