Senior / Staff ML Onboard Optimization Engineer
Core
Optimizing deep learning models for inference on onboard compute systems in autonomous trucks and robotaxis.
Role type
Senior/Staff ML Onboard Optimization Engineer
Builds
Optimized inference pipelines for autonomous vehicle onboard compute systems
Domain
Autonomous transportation / Physical AI / Embedded Systems
Deliverable
production ML models
Required skills
CUDA kernel development, model profiling (CPU/GPU), TensorRT, PyTorch, C++, Python, Rust, embedded systems integration
Preferred skills
Bazel build systems, model compilation/exporting, third-party package integration
Technologies
TensorRT, modelopt, PyTorch Profiler, NVIDIA Nsight, Nvidia Jetson, Nvidia Thor
Responsibilities
Collaborate with autonomy engineers to scale self-driving systems; Expand model deployment pipelines to new GPUs and embedded systems; Create and benchmark new CUDA kernels; Profile model runtime and memory to identify performance bottlenecks
Seniority
Senior/Staff, hands-on IC
