17h ago

Senior ML Training Optimization Engineer

Remote US & Canada

โœจ $180k-$250k / yearest.

full-timesenior Remotetransportation

๐Ÿ›  Tech Stack

๐Ÿ’ผ About This Role

You'll collaborate with autonomy engineers to scale self-driving systems using an AI-first approach, expanding model deployment pipelines to new GPUs and embedded systems. Your work will directly optimize inference on autonomous trucks.

๐ŸŽฏ What You'll Do

  • Optimize models using TensorRT and modelopt for inference on GPUs
  • Create and benchmark custom CUDA kernels for inference
  • Profile runtime and memory to identify performance bottlenecks
  • Expand deployment pipeline to new platforms like Jetson or Thor

๐Ÿ“‹ Requirements

  • 6+ years industry experience in CS, Robotics, or related field
  • Proficiency in Python, C++, or Rust
  • Experience with deep learning frameworks like PyTorch
  • Skilled in profiling using PyTorch Profiler and NVIDIA Nsight

โœจ Nice to Have

  • Experience with model compilation and TensorRT
  • Implementing custom CUDA kernels
  • Experience with Bazel build systems

๐ŸŽ Benefits & Perks

  • ๐Ÿ–๏ธ Unlimited PTO
  • ๐Ÿฅ Health insurance
  • ๐Ÿ“ˆ Stock options
  • ๐Ÿ’ป Remote work

๐Ÿ“จ Hiring Process

Estimated timeline: 2-4 weeks ยท AI estimate

  1. 1Recruiter Screenยท 30 min
  2. 2Technical Interviewยท 60 min
  3. 3On-site Interviewยท 3 hours
0 0 0