Lead ML Training Optimizer Scalable AI Systems
Listed on 2026-10-12
-
Software Development
AI Engineer (Applied/Software), Machine Learning/ ML Engineer
Waabi is seeking a highly skilled engineer to develop distributed training frameworks and optimize performance for research and production environments. You will profile runtime/memory, identify new technologies such as CUDA kernels and quantization, and collaborate with researchers to implement best practices for efficient resource usage.
You will also design and improve tooling and dashboards to promote broad adoption across teams, contributing to Waabi’s cutting-edge autonomous driving
This role, Lead ML Training Optimizer for Scalable AI Systems at Waabi, could be your next move.
Please review the full job details above before applying.
If your experience matches this role, we encourage you to apply.
All applications are reviewed carefully by our team.
The position is based in Phoenix, AZ, United States.
This opportunity is part of our work in Engineering.
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).