over 1 year ago
Remote, Americas +5 moreSenior / Staff+
Base Salary
$141k - $249k/yr
Responsibilities
- Collaborate with autonomy and algorithm engineers to scale safe self-driving systems.
- Expand the model deployment pipeline to new GPUs and embedded systems for onboard compute.
- Optimize truck-deployed models using TensorRT and ModelOpt.
- Create and benchmark CUDA kernels for inference.
- Profile model runtime, CPU/GPU code, and memory usage to identify performance bottlenecks.
Requirements
- MS/PhD or a bachelor’s degree in Computer Science, Robotics, or a similar technical field, with a minimum of six years of industry experience for the bachelor’s-degree path.
- Strong coding proficiency in Python, C++, or Rust.
- Experience with deep learning frameworks such as PyTorch.
- Experience 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.
- Collaborative mindset and willingness to help others.
- Passion for self-driving technology and solving complex technical problems.
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 the office.
- Regular team-building activities and social events, both onsite and virtually.
- Access to an onsite facility with a gym, games room, collaborative workspaces, and patio.
- The role is available across Waabi’s US locations, including offices in San Francisco, Dallas, and Pittsburgh.
About Waabi
Waabi builds AI-driven self-driving technology for long‑haul freight, centered on the Waabi Driver autonomous trucking system and the Waabi World simulator. The company partners with carriers and vehicle makers to commercialize autonomous trucks and explores robotaxi applications. Founded in 2021 by Raquel Urtasun, Waabi is privately held, headquartered in Toronto, with offices in San Francisco, Dallas, and Pittsburgh.