PnP Optimization Engineer

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's powering commercial autonomous trucks and robotaxis. Waabi is backed by and partners with world leaders in AI, automotive, logistics, and deep tech. With offices in Toronto, San Francisco, Dallas, and Pittsburgh, Waabi is growing quickly and looking for diverse, innovative and collaborative candidates who want to impact the world in a positive way. To learn more visit: www.waabi.ai You will… - Collaborating with research scientists during algorithm design to ensure code is efficiently designed from inception - Identify and communicate best practices for model development - Reformulate performance bottlenecks - Develop a framework suitable for mixed precision training of multi-task, multi-modality models in a heterogeneous distributed training environment - Research, implement, and test alternative formulations for fundamental DNN operations and AV centric representations - Enable safe and efficient deployment of PnP models Qualifications: - PyTorch API and implementation - CUDA system design - Asynchronous programming model - Heterogeneous compute - C++ / libtorch Bonus: - Machine learning model architecture design - TensorRT - CUDA kernel implementation - Experience with autonomous vehicle multi-sensor model architectures