← All roles

Senior / Staff ML Onboard Optimization Engineer

Waabi· Software – Autonomy & Algorithms
Apply on company site
Hybrid (Remote US & Canada; Dallas, TX; Phoenix, AZ; Pittsburgh, PA; San Francisco, CA; Toronto, ON)$141k - $249kFull-timePosted 7mo ago

About the role

Senior / Staff ML Onboard Optimization Engineer role focused on scaling safe self-driving systems, expanding deployment pipelines to GPUs and embedded systems, and optimizing models for onboard compute in autonomous vehicles.

Responsibilities

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

Skills

Required:PythonC++PyTorchCUDAProfiling toolsNvidia embedded platformsCollaboration
Nice to have:Rust

Benefits

  • 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.
  • World-class facility that includes a gym, games room, multiple collaborative working spaces and a patio (when in office).
Last updated Oct 21, 2026. Interested? Apply on the company site.

About Waabi

pioneering Physical AI, starting with autonomous trucks.

Waabi is pioneering Physical AI, starting with autonomous trucks. We developed a next-generation approach leveraging an end-to-end interpretable and verifiable AI model that’s powered by the industry's most realistic neural simulator. This dramatically reduces the development time and resources needed to bring self-driving vehicles to public roads — safely and at scale.

Autonomous trucksPhysical AIneural simulatorAI modelself-driving vehicles
Company profileWebsite

More open roles at Waabi