Senior AI Tooling Engineer

Jobtailor; Sunnyvale (CA); On-site; USD 140,000 - 190,000; Full time

Job summary

Jobtailor in Sunnyvale seeks an experienced AI/ML engineer to enhance training and inference efficiency in large-scale models. You will build workflows for correctness and performance and create tooling to predict model behavior from architecture and data shapes.

Proficiency with PyTorch, TensorFlow, TensorRT and NVIDIA Nsight tools is required, as is production-grade Python/C++ coding. You will influence architecture decisions within GM and help evolve the AI toolchain.

Qualifications

• 3+ years of AI/ML experience.
• Experience with PyTorch and TensorFlow.
• Proficiency writing production-grade Python/C++ code; SDLC.
• BS or higher in CS/CE/EE, or equivalent.

Responsibilities

• Identify opportunities to improve training and inference efficiency.
• Build workflows for correctness and performance on in-car compute and sensors.
• Develop tooling to predict model performance from architecture and data shape.
• Trace performance on large distributed training/distillation on GPUs.
• Evolve toolchain to leverage advancements in AI.
• Influence model architecture decisions within GM.

Skills

Python

C++

AI/ML

Production code

SDLC

Education

Bachelor's degree in CS/CE/EE

Tools

PyTorch

TensorFlow

TensorRT

Nsight-systems

Nsight-compute

Job description

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

Requirements

• 3+ years of experience in the field of AI/ML
• Experience with ML frameworks (e.g., PyTorch, TensorFlow) 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

Core Competencies

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 toolchains to leverage advancements in AI.

Tools & Technologies

• PyTorch
• TensorFlow
• TensorRT
• Nsight-systems
• Nsight-compute