Job Description
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 training frameworks for resear...
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 training frameworks for resear...
Apply for This Position
Ready to take the next step? Click the button below to submit your application.
Submit Application