Senior / Staff Ml Optimization Engineer

Waabi

Remote, US
Base: $141,000 - $249,000; bonus/equity: equity in...
Fully remote
Distributed training framework development
Model runtime and memory profiling
Python c++ or rust coding proficiency
Waabi is seeking a Senior/Staff ML Optimization Engineer to join their remote team, focusing on developing advanced technologies for autonomous transportation. The ideal candidate will have extensive experience in machine learning, coding proficiency, and a passion for self-driving technologies. The position offers a competitive salary and numerous benefits in a rapidly growing company that values innovation and collaboration

Job Summary

  • Waabi is unlocking the next era of autonomous transportation with technology powering commercial autonomous trucks and robotaxis.
  • The role involves building distributed training frameworks to drive training towards new levels of stability and efficiency.
  • Candidates must be skilled in profiling CPU and GPU code using tools such as PyTorch Profiler and NVIDIA Nsight.

Matching Summary

Match Score: 85

Waabi is seeking a Senior/Staff ML Optimization Engineer to join their remote team, focusing on developing advanced technologies for autonomous transportation. The ideal candidate will have extensive experience in machine learning, coding proficiency, and a passion for self-driving technologies. The position offers a competitive salary and numerous benefits in a rapidly growing company that values innovation and collaboration.

Salary

Base: $141,000 - $249,000; Bonus/Equity: Equity incentive awards and annual performance bonus included; Benefits: Competitive perks & benefits

Skills & Requirements

Must-have

  • Distributed training framework development
  • Model runtime and memory profiling
  • Python C++ or Rust coding proficiency
  • Deep learning frameworks experience PyTorch
  • CPU and GPU code profiling skills

Nice-to-have

  • Experience with TensorRT and model compilation
  • Custom CUDA kernel implementation skills
  • Bazel build system integration experience
  • Collaborative team player mindset
  • Passion for self-driving technology

Key Requirements

  • MS/PhD or Bachelors degree with 6 years industry experience
  • Solid coding proficiency in Python C++ or Rust
  • Experience across data processing distributed training and model deployment

Work Rights

Not specified

Tailored Resume

Cover Letter