Waabi1d ago

New

Senior / Staff ML Training Optimization Engineer

Us & CanadaFull-timesenior

Optimization EngineerData & AI

0 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 4 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

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

- Build standardized 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. Examples include designing new CUDA kernels, quantization-aware training and inference, and compilation/deployment techniques.

- Work with researchers and ML engineers on best-practices for optimal resource usage.

- Create and improve tooling and dashboards to ensure broad adoption of your work.

Qualifications:

- MS/PhD or Bachelors degree with a minimum of 4 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 or Jax.

- 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 identifying when custom CUDA kernels are needed, and implementing them.

- Experience in Bazel in a monorepo environment, and integrating third party packages into dev environments.

- Experience with Kubernetes-based training platforms.

Location & Eligibility

Where is the job

Us & Canada

Hybrid — some on-site time required

Who can apply

Same as job location

Listing Details

Posted: May 8, 2026
First seen: May 8, 2026
Last seen: May 9, 2026

Posting Health

Days active: 0
Repost count: 0
Trust Level: 70%
Scored at: May 8, 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

Jobs

External application · ~5 min on Waabi's site

Please let Waabi know you found this job on Jobera.