Senior / Staff ML Training Optimization Engineer (Toronto)

About Waabi

Waabi, founded by AI visionary Raquel Urtasun, is the leader in Physical AI. With a world‑class team, we’re unlocking the next era of autonomous transportation with technology that powers commercial autonomous trucks and robotaxis. Waabi is backed by and partners with world leaders in AI, automotive, logistics, and deep tech.

Role Overview

You will build standardized distributed training frameworks for research and production, drive our training toward 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, and work with researchers and ML engineers on best‑practices for optimal resource usage. You will also create and improve tooling and dashboards to ensure broad adoption of your work.

Responsibilities

• Build and maintain distributed train...