AI Optimization Engineer at Waabi

Waabi • mississauga, peel region, Canada • Posted June 05, 2026

Location mississauga, peel region
Job Type Full-time
Category Engineering
Posted June 05, 2026
Elevate self-driving tech with Waabi as an AI Optimization Engineer. Leverage your expertise in deep learning frameworks, coding, and AI-driven systems in a collaborative, innovative environment.

Waabi seeks candidates to enhance self-driving systems through an AI-first methodology. This senior role emphasizes collaboration with autonomy and algorithm engineers, developing standardized distributed training frameworks. Your contributions will focus on optimizing model runtimes, discovering performance bottlenecks, and incorporating emerging technologies into our training and inference frameworks.

Key Responsibilities:
• Collaborate with engineers to scale self-driving systems
• Develop standard frameworks for training and production
• Profile model runtimes to identify bottlenecks
• Evaluate new technologies for training frameworks
• Implement efficient CUDA kernels for enhanced performance

Requirements:
• MS/PhD...

Interested in this role?

Click the button below to start your application.

Apply Now