[ engineering ]

Senior / Staff Ml Onboard Optimization Engineer

Waabi
Location
Remote
Work mode
Remote
Type
full time
Listed

Applications go directly to Waabi. Ramai takes no fee for placement and no money for ranking.

[ About this role ]

Senior/Staff ML Onboard Optimization Engineer at Waabi. This role involves collaborating with autonomy engineers to scale self-driving systems, expanding the model deployment pipeline to new hardware, and optimizing models using frameworks like TensorRT. Responsibilities include creating CUDA kernels for inference and profiling model runtime and memory.

Skills and focus areas

  • python
  • c++
  • rust
  • deep learning frameworks
  • pytorch
  • profiling cpu and gpu code
  • pytorch profiler
  • nvidia nsight
  • nvidia embedded platforms
  • nvidia jetson
  • nvidia thor
  • cuda kernel development

Summary written by Ramai from the original posting. Full details and requirements are on Waabi’s careers page.

[ More at Waabi ]

See all 92 open roles at Waabi →

[ Similar roles at other startups ]

Browse all open engineering roles →

[ Not quite right? ]

Tell the assistant what you’re after — stage, location, stack, visa sponsorship — and it searches every open role for you. A few questions are free, no account needed.