Waabi12mo ago

$141,000 – $249,000/yr

Senior / Staff ML Onboard Optimization Engineer

Us & CanadaFull-timesenior

OtherOptimization EngineerData & AI

2 views0 saves0 applied

Apply Now

Quick Summary

Overview

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.

Requirements Summary

- MS/PhD or Bachelors degree with a minimum of 6 years of industry experience in Computer Science, Robotics and/or similar technical field(s) of study. - Solid coding proficiency in a variety of coding languages including Python, C++ or Rust.

Technical Tools

cpppythonpytorchrustdeep-learning

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...

- Collaborate closely with autonomy and algorithm engineers to scale safe self-driving systems using an AI-first approach.

- Expand the model deployment pipeline to new GPUs and embedded systems for the next generation of our onboard compute system.

- Use frameworks such as TensorRT and modelopt to optimize the models running on the truck.

- Create and benchmark new CUDA kernels for inference.

- Comprehensively profile model runtime and memory to pinpoint performance bottlenecks.

Qualifications:

- MS/PhD or Bachelors degree with a minimum of 6 years of industry experience in Computer Science, Robotics and/or similar technical field(s) of study.

- Solid coding proficiency in a variety of coding languages including Python, C++ or Rust.

- Experience in deep learning frameworks such as PyTorch.

- Skilled in profiling CPU and GPU code using tools such as PyTorch Profiler and NVIDIA Nsight.

- Experience with Nvidia embedded platforms such as Nvidia Jetson or Thor.

- Open-minded and collaborative team player with willingness to help others.

- Passionate about self-driving technologies, solving hard problems, and creating innovative solutions.

Bonus/nice to have:

- Experience in model compilation and exporting, interaction with lower level concepts like TensorRT.

- Experience in identifying when custom CUDA kernels are needed, and implementing them.

- Experience in Bazel build systems, and integrating third party packages into dev environments.

Location & Eligibility

Where is the job

Us & Canada

Hybrid — some on-site time required

Who can apply

Same as job location

Listed under

Worldwide

Listing Details

Posted: April 24, 2025
First seen: March 26, 2026
Last seen: May 14, 2026

Posting Health

Days active: 49
Repost count: 0
Trust Level: 44%
Scored at: May 14, 2026

Signal breakdown

freshnesssource trustcontent trustemployer trust

Apply for this position

Waabi

lever

Waabi is an AI company developing generative AI-powered autonomous driving technology for long-haul trucking, aiming for commercial deployment in 2025.

Employees

125

Founded

2021

Domain

waabi.ai

Jobs

View 64 jobs

View company profile

Salary

$141,000 – $249,000

per year