PitchMeAI Jobs
Waabi

Senior / Staff ML Onboard Optimization Engineer

Waabi · Remote US & Canada

Posted about 3 months ago · $141,000 - $249,000

or apply directly on Waabi's site. We never take the application ourselves.

Is this posting real?

This role has been open
79 days
Waabi's roles stay open a median of 80 days
Reposted
No
Salary listed
Yes
65% of Waabi's roles list one
Ghost-job risk at Waabi
high
81 stale, 2 reposted of 88 open
Hiring momentum
97 roles opened in the last 90 days
↑ up vs. the prior 90 days
Last confirmed on the employer's board
2026-10-23

Measured from postings appearing on and disappearing from Waabi's own lever board since 2026-08-04. Full hiring picture for Waabi.

About this role

In this role, you will work closely with autonomy and algorithm engineers to enhance self-driving systems through an AI-first approach. Your responsibilities will include expanding the model deployment pipeline, optimizing models using frameworks like TensorRT, and profiling model performance to identify bottlenecks. You'll also create and benchmark CUDA kernels for inference on the onboard compute system.

Our read on this posting3.8out of 5
benefits
4/5
freshness
1/5
career value
5/5
role clarity
4/5
pay transparency
5/5

Scored from the posting itself — how clearly the role is described, how much it says about pay and benefits, and how recently it was listed. Not a judgement of Waabi as an employer.

What you need

  • MS/PhD or Bachelor's degree with a minimum of 6 years of industry experience in Computer Science, Robotics or similar technical field(s)
  • Solid coding proficiency in 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

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

What you get

  • Competitive compensation and equity awards
  • Health and Wellness benefits encompassing Medical, Dental and Vision coverage
  • 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

Worth weighing

  • No specific mention of the tech stack beyond CUDA and deep learning frameworks
  • Salary range is broad and may vary significantly based on experience
  • Role involves complex problem-solving in a rapidly evolving field, which may require continuous learning

Summarised from Waabi's posting. Read the full original.

Listed by Waabi on their lever job board, last confirmed open on 2026-10-23. PitchMeAI is not the employer.

More roles at Waabi

All 88 open roles at Waabi →