PnP Optimization Engineer

Summary

Join Waabi, a leading AI company revolutionizing self-driving technology, and contribute to the development of cutting-edge algorithms. Collaborate with research scientists, identify and implement best practices for model development, and optimize performance. Develop frameworks for mixed-precision training and research alternative formulations for DNN operations. Enable safe and efficient deployment of plug-and-play models. Waabi offers competitive compensation, equity awards, and a comprehensive benefits package. The company fosters a diverse and inclusive work environment.

Requirements

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

Responsibilities

• 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

Preferred Qualifications

• Machine learning model architecture design
• TensorRT
• CUDA kernel implementation
• Experience with autonomous vehicle multi-sensor model architectures

Benefits

• Competitive compensation and equity awards
• Health and Wellness benefits encompassing Medical, Dental and Vision coverage (for full-time employees only)
• Unlimited Vacation
• Flexible hours and Work from Home support
• Daily drinks, snacks and catered meals (when in office)
• Regularly scheduled team building activities and social events both on-site, off-site & virtually