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
- 1Recruiter Screenยท 30 min
- 2Technical Interviewยท 60 min
- 3On-site Interviewยท 3 hours
0 0 0