Job Description

Waabi is seeking a PNP Optimization Engineer to collaborate on algorithm design, identify performance bottlenecks, and develop frameworks for training and deploying AI models for self-driving technology. The role involves research, implementation, and testing of DNN operations and AV-centric representations.

Responsibilities

• Collaborate on algorithm design for efficient code
• Identify and communicate best practices for model development
• Reformulate performance bottlenecks
• Develop framework for mixed precision training
• Research and implement DNN operations
• Enable safe and efficient deployment of models

Requirements

• PyTorch API and implementation
• CUDA system design
• Asynchronous programming model
• Heterogeneous compute
• C++ / libtorch

Benefits

• No benefits