Senior / Staff ML Optimization Engineer

Us / Canada · Remote (us/canada)Full-timesenior

OtherOptimization EngineerData & AI

0 views0 saves0 applied

Apply Now

Quick Summary

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.

Technical Tools

OtherOptimization EngineerData & AI

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

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

- Build distributed training frameworks for research and production, drive our training towards new levels of stability and efficiency.

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

- Identify and evaluate emerging technologies that can be adopted into Waabi’s training and inference frameworks, including but not limited to efficient CUDA kernels for training, quantization, model exporting and compilation for inference.

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.

- Experience across different stages of the development pipeline: data processing, distributed training and model deployment.

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

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

The US yearly salary range for this role is: $141,000 - $249,000 in addition to competitive perks & benefits. Waabi’s yearly salary ranges are determined based on several factors in accordance with the Company’s compensation practices. The salary base range is reflective of the minimum and maximum target for new hire salaries for the position across all US locations. Note: The Company provides additional compensation for employees in this role, including equity incentive awards and an annual performance bonus.

Listing Details

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

Posting Health

Days active: 29
Repost count: 0
Trust Level: 44%
Scored at: April 24, 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 61 jobs

View company profile

Salary

$141,000 – $249,000

per year