Senior AI/ML Tooling Engineer

Position Overview

As a Senior AI/ML Tooling Engineer, you will build and enhance GM’s internal ML tooling for high performance software. You will leverage state of the art tools like Nsight Systems and PyTorch to analyze and optimize the distillation, training, and inference of ML models that are critical to the Autonomous Vehicle software stack. This role involves close collaboration with engineers and researchers across various AV Engineering teams as well as AV hardware teams.

Key Responsibilities

• Identify new opportunities to improve both training and inference efficiency.
• Build workflows for correctness and performance analysis on physical in-car compute and sensor systems.
• Develop tooling to predict model performance based on architecture and data shape.
• Construct tools to trace actual performance on large distributed training and distillation jobs running on the world’s most powerful GPUs, and analyze the results.
• Continually evolve the toolchain and stack to leverage the latest advancements in AI.
• Influence model architecture decisions and strategy within GM.

Required Qualifications

• 5+ years of experience in the field of AI/ML.
• Proficiency with ML frameworks (e.g., PyTorch, TensorFlow) and the NVIDIA developer ecosystem (TensorRT, Nsight-systems, Nsight-compute).
• Expertise in writing production-quality Python/C++ code.
• Strong understanding of the software development life-cycle including coding, debugging, optimization, testing, and integration.
• Bachelor’s degree or higher in CS/CE/EE, or equivalent experience.

Preferred Qualifications

• Experience developing and deploying machine learning models.
• Proficiency in GPU programming (CUDA) and familiarity with ML software stacks (e.g., cuDNN, cuBLAS).
• Experience with ML accelerators and hardware architecture.