Sr. Manager, Hardware, Machine Learning/AI
Role Overview
Lead hardware-aware machine learning infrastructure and model deployment for Rivian edge AI platforms.
Shape the long-term direction of the edge AI stack, including support for increasingly large and complex neural network models.
Key Skills for This Role
Full Job Posting
Role Overview
Lead hardware-aware machine learning infrastructure and model deployment for Rivian edge AI platforms.
Shape the long-term direction of the edge AI stack, including support for increasingly large and complex neural network models.
Core Responsibilities
- Direct machine learning enablement workflows for edge AI platforms.
- Oversee front-end compiler and model-ingestion tooling roadmaps for current and future platforms.
- Drive architecture for quantization-aware training and model optimization infrastructure at scale.
- Manage model porting, optimization, and benchmarking for hardware accelerators and NPUs.
- Lead hardware and systems initiatives addressing memory, scheduling, and data-movement constraints.
- Establish standards for evaluating model fidelity and deployment quality across execution paths.
- Align technical execution with product milestones in collaboration with executive leadership.
- Recruit, mentor, and grow an engineering team focused on model deployment, compiler infrastructure, and hardware-aware ML.
- Shape the long-term direction of the edge AI stack.
Required Qualifications
- Proven experience leading or managing teams in machine learning systems, model deployment, or AI infrastructure.
- Strong experience optimizing models for edge or accelerated inference, including quantization and performance tuning.
- Experience building or extending compiler, toolchain, or graph-transformation infrastructure for ML workloads.
- Strong understanding of hardware/software co-design for inference acceleration.
- Experience with low-power edge platforms, NPUs, custom accelerators, GPUs, or related architectures.
- Ability to reason about memory allocation, execution scheduling, bandwidth constraints, and system performance.
- Experience translating research concepts into practical tooling and production workflows.
- Exceptional cross-functional leadership and a history of technical and strategic impact.
Preferred Qualifications
- Experience with quantization-aware training, post-training quantization, compression, pruning, or sparsity optimization.
- Experience deploying vision or multimodal models.
- Familiarity with ONNX and model graph transformation pipelines.
- Experience validating numerical behavior across floating-point and integer execution paths.
- Experience supporting external customers, partner enablement, or AI platform productization.
- Background in both model-level and hardware-level optimization.
- Startup or founder-level experience building end-to-end AI platform capabilities.
Apply for this job in 1 click
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
More jobs at Rivian
Sr. Inventory Analyst
Irvine, USA
Rivian seeks a Senior Inventory Analyst to optimize inventory, working capital, and service-center fill rates across its supply chain. The role calls for 5+ years in supply chain or inventory planning, expertise in SAP a
Field Service Technician - Night Shift - Denver, CO
Denver, USA
Early Career - Software Engineer, Enterprise Software
Atlanta, USA
Product Guide III (Part-time) - Corte Madera Showroom
Corte Madera, USA
Field Recruiter, Sales & Delivery (Contract)
Seattle, USA
Staff Technical Program Manager, Vehicle Service Readiness & Product Integration
Irvine, USA
Sr. Inventory Analyst
Irvine, USA
Field Service Technician - Night Shift - Denver, CO
Denver, USA
Early Career - Software Engineer, Enterprise Software
Atlanta, USA
Technician, Paint Mix Room ( 2nd Shift)
Normal, USA
Product Guide III (Part-time) - Corte Madera Showroom
Corte Madera, USA
Lead, Design, Commercial Facilities
New York City, USA
Field Recruiter, Sales & Delivery (Contract)
Seattle, USA
Staff Technical Program Manager, Vehicle Service Readiness & Product Integration
Irvine, USA