Base Career helps you apply smarter for this job.
Key skills for this role
Optimize large-scale models (Multi-Modal Sensor Fusion models, LLMs, VLMs) using advanced quantization (PTQ, QAT), pruning, mixed-precision inference frameworks, and parameter-efficient fine-tuning (LoRA, QLoRA).
Architect and implement model conversion and compilation pipelines using TensorRT for edge deployment.
Perform rigorous parity checking, accuracy recovery, and latency benchmarking between PyTorch frameworks and compiled edge binaries.
Develop and optimize custom ML OPs and TensorRT Plugins with efficient CUDA kernels to minimize latency and maximize memory bandwidth on AI accelerators.
Write production-level, low latency, and memory-safe C++ and CUDA code for real-time inference on vehicle systems.
Familiarity with SOTA autonomous driving perception algorithms (temporal 3D object detection, BEV, 3D Occupancy Networks) and multi-modal sensor processing (Vision, LiDAR, Radar).
Skip the repetitive application forms
Install the Base Career Chrome Extension and autofill job applications across major job boards with your profile.
Trusted by over 500,000 job seekers on Base Career
More from this employer
Foster City, USA
Foster City, USA
Foster City, USA
Foster City, USA
Las Vegas, USA
Hayward, USA
Foster City, USA
Foster City, USA
Foster City, USA
Foster City, USA
Foster City, USA
Foster City, USA
Experience with distributed training pipelines and model/tensor parallelism (PyTorch Distributed, Ray, DeepSpeed, Megatron-LM) and runtime efficiency optimization for GPU clusters.
Experience with end-to-end autonomous driving paradigms (VLM/VLA models, Foundation models) and edge deployment technologies (e.g., TensorRT-LLM).
Zoox, an Amazon subsidiary, develops purpose-built autonomous robotaxis designed from the ground up for rider comfort without a traditional driver's seat.
Full-time
Senior
Hybrid
Apply faster on company sites with our extension.