Apply

Ready to go for it?

AI Apply speeds things up—apply directly if you prefer.

FREE ACCESS
5,000–10,000 jobs/day
JobTailor Logo

See all jobs on JobTailor

Search thousands of fresh jobs every day.

Discover
  • Fresh listings
  • Fast filters
  • No subscription required
Create a free account and start exploring right away.
Waabi

Senior/Staff ML Training Optimization Engineer

Waabi

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

Posted 5/8/2026full-timeRemote US & Canada • 🇨🇦 CanadaSenior💰 $141,000 - $249,000 per yearWebsite

Tech Stack

Tools & technologies
PythonPyTorchRust

About the role

Key responsibilities & impact
  • 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.

Requirements

What you’ll need
  • 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.
  • 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.
  • 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.

Benefits

Comp & perks
  • 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 (ping pong table, video game consoles, board games, etc), multiple collaborative working spaces and a gorgeous patio!(when in office)

ATS Keywords

✓ Tailor your resume
Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools
PythonC++Rustdeep learningCUDAmodel optimizationmodel profilingmodel compilationNVIDIA NsightBazel
Soft Skills
collaborativeteam playeropen-mindedproblem-solvinginnovative
Certifications
MSPhDBachelor's degree