Senior / Staff ML Onboard Optimization Engineer

The role

You will work with autonomy and algorithm engineers to scale self-driving systems. Your focus is optimizing the machine learning models that run on the truck, across new GPUs and embedded systems.

What you will do

• Collaborate closely with autonomy and algorithm engineers to scale safe self-driving systems.
• Expand the model deployment pipeline to new GPUs and embedded systems for the next generation of onboard compute.
• Use frameworks such as TensorRT and modelopt to optimize models running on the truck.
• Create and benchmark new CUDA kernels for inference.
• Profile model runtime and memory to find performance bottlenecks.

Required qualifications

• MS/PhD or Bachelor's degree with at least 6 years of industry experience in Computer Science, Robotics or a similar technical field.
• Solid coding proficiency in Python, C++ or Rust.
• Experience with deep learning frameworks such as PyTorch.
• Skilled in profiling CPU and GPU code with tools such as PyTorch Profiler and NVIDIA Nsight.
• Experience with NVIDIA embedded platforms such as Jetson or Thor.
• Open-minded and collaborative team player who helps others.
• Passionate about self-driving technologies and solving hard problems.

Nice to have

• Experience in model compilation and exporting, and interaction with lower-level concepts like TensorRT.
• Experience identifying when custom CUDA kernels are needed and implementing them.
• Experience with Bazel build systems and integrating third-party packages into dev environments.

Benefits

• Competitive compensation and equity awards.
• Medical, Dental and Vision coverage.
• Unlimited vacation.
• Flexible hours and work from home support.
• Daily drinks, snacks and catered meals when in office.
• Regular team building activities and social events on-site, off-site and virtually.
• World-class facility with a gym, games room and collaborative workspaces when in office.

Before you apply

• Salary range applies to US locations.

About the company

Waabi is a Physical AI company founded by Raquel Urtasun. It builds technology for commercial autonomous trucks and robotaxis.