Senior ML Engineer - Model Compression

Job Description

The Compression and Parity team in GM’s Autonomous Vehicle (AV) Organization enables repeatable, high‑velocity model deployments through principled and automated model compression under strict safety guarantees. We partner closely with model developers, deployment, and infra engineers to ship numerically robust, low‑latency models to the car, blending rigorous analysis with state‑of‑the‑art methods and our own innovations.

• Develop and iterate on quantization and compression strategies for AV models, considering numerical properties, safety and latency constraints, and hardware performance, and partner on deployment of quantized models to NVIDIA‑based AV hardware with deployment, compiler, and kernel teams.
• Advance numerical sensitivity analyses to recommend safe compression policies per op/layer/block, use AV‑relevant metrics (perception, trajectory, etc.) to evaluate compressed models, and collaborate with Embodied AI to support compression‑aware modeling.
• Evolve sensitivity analysis, compression, and parity tooling into a connected, automated flow that makes low‑precision deployments repeatable, reliable, and low‑touch, emphasizing robust execution and maintainability.
• Bridge the gap between state‑of‑the‑art model compression research and safety‑constrained deployment while making strong technical contributions in cross‑functional projects and educating others on best practices.

• Bachelor’s degree in Computer Science, Electrical Engineering, Physics, Mathematics, Data Science / ML, or a closely related quantitative field (or equivalent experience).
• 3+ years of industry experience focused on model optimization and deployment, with significant hands‑on work in neural network quantization, model compression, or efficient inference.
• Strong proficiency in PyTorch and experience with graph‑level representations (e.g., PyTorchFX, ONNX) for capture and manipulation.
• Background in numerical linear algebra and optimization (conditioning, spectral properties, Jacobians, Hessians) and how they relate to quantization robustness.

• Master’s or PhD degree in related quantitative fields.
• Deep experience with PTQ and QAT, compression frameworks (e.g., PT2E, Model Opt, torchao) and advanced quantization algorithms (e.g., GPTQ, AWQ, Smooth Quant, QuIP, SparseGPT), as well as with building or extending quantization tool chains.
• Hands‑on experience designing numerics observability and sensitivity tooling integrated into training or evaluation pipelines (logging ranges, saturation, quant noise, etc.).
• Track record of collaboration, including leading cross‑functional initiatives and mentoring others.
• Experience with additional compression techniques such as structured/unstructured pruning, low‑rank decomposition, or knowledge distillation.
• Experience with perception and/or transformer‑based models (e.g., multi‑view encoders, BEV backbones, detection/segmentation heads, trajectory or planning networks), ideally in AV / ADAS.
• General understanding of kernel performance and optimization for reduced precision formats.
• Direct experience with specialized hardware accelerators for edge deployment on tight latency and memory budgets (automotive SoCs, robotics platforms, or similar).
• Published research, open‑source contributions, or other notable intellectually curious work in quantization, compression, or efficient inference.
• 3+ years of industry experience focused on model optimization and deployment, with significant hands‑on work in neural network quantization, model compression, or efficient inference.

• Salary range: $128,700 to $261,300. The actual base salary will vary based on applicability.
• Bonus potential through incentive pay program based on company performance, job level, and individual performance.
• Health and wellbeing benefits including medical, dental, vision, health savings account, flexible spending accounts, retirement savings plan, sickness and accident benefits, life insurance, paid vacation & holidays, tuition assistance programs, employee assistance program, GM vehicle discounts, and more.

This role is…