Waabi · 4 days ago
PnP Optimization Engineer
Canada, KY
Full-time
Hybrid
Mid, Senior Level
$158K/yr - $269K/yrWaabi is an AI company building the next generation of self-driving technology. The PnP Optimization Engineer will collaborate with research scientists on algorithm design, identify best practices for model development, and enable the efficient deployment of PnP models.
Artificial Intelligence (AI)AutomotiveAutonomous VehiclesMachine Learning
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
Qualification
Required
PyTorch API and implementation
CUDA system design
Asynchronous programming model
Heterogeneous compute
C++ / libtorch
Preferred
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.
Company
Waabi
Waabi is pioneering Physical AI starting with autonomous trucking.
Founded in 2021
51-200 employees
H1B Sponsorship
Waabi has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (8)
2024 (6)
2023 (6)
2022 (4)
Funding
Current Stage
Growth StageTotal Funding
$283.04MKey Investors
Volvo Group Venture CapitalKhosla Ventures
2024-06-18Series B· $200.48M
2023-01-18Series Unknown
2021-06-08Series A· $82.55M
Leadership Team
Raquel Urtasun
Founder & CEO
Lior Ron
COO
Recent News
Calgary Herald
2026-01-08
2025-12-13
Company data provided by crunchbase