🌿 Back to all jobs

🥝 AI ML Optimization Engineer at Waabi

Waabi | vancouver, Canada | Posted June 05, 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:...

Apply for This Position

Submit Application