Senior AI Tooling Engineer

Responsibilities

• Identify new opportunities to improve both training and inference efficiency
• Build workflows for correctness and performance analysis on physical in-car compute and sensors
• Build tooling to predict model performance based on architecture and data shape
• Build tooling 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

• 3+ years of experience in the field of AI/ML
• Experience with ML frameworks (e.g., PyTorch, Tensor Flow) and NVIDIA developer ecosystem (TensorRT, Nsight-systems, Nsight-compute)
• Expertise in writing production quality Python/C++ code
• Expertise in the software development life‑cycle - coding, debugging, optimization, testing, integration
• BS, or higher degree, in CS/CE/EE, or equivalent

Demonstrates expertise in AI/ML with a strong focus on improving training and inference efficiency, alongside proficiency in ML frameworks and production‑quality coding. Capable of influencing model architecture decisions and evolving tool chains to leverage advancements in AI.

• Py Torch
• Tensor Flow
• TensorRT
• Nsight-systems
• Nsight-compute