Machine Learning Training Engineer at Waabi

Waabi • winnipeg, mb, Canada • Posted May 30, 2026

Location winnipeg, mb
Job Type Full-time
Category Other-General
Posted May 30, 2026
Transform the future of transportation with Waabi as a Machine Learning Training Optimization Engineer. Drive efficiency and stability in training frameworks for autonomous trucks and robotaxis.

At Waabi, an innovator in Physical AI, you'll be part of a talented team pushing boundaries in autonomous technology. This role requires building standardized training frameworks and identifying performance bottlenecks in models. Collaboration with ML engineers will also be essential to optimize resource usage and improve tooling for wider adoption.

Key Responsibilities: • Build and standardize distributed training frameworks • Profile model runtime and memory for performance insights • Evaluate emerging technologies for training enhancements • Collaborate with researchers on optimal resource usage • Create dashboards and tooling for broader adoption

Requirements: • MS/PhD or Bachelor's with 4+ years in a technical field • Proficient in Python, C++, or Rust • Familiar...

Interested in this role?

Click the button below to start your application.

Apply Now