Senior AI Performance Architect for Autonomous Systems

NVIDIA Corporation; Santa Clara (CA); On-site; USD 184,000 - 357,000; Full time

Job summary

NVIDIA Corporation in Santa Clara, California, is seeking a Solutions Architect with a performance engineering background to accelerate Physical AI workloads in Autonomous Vehicles and Robotics using NVIDIA's full-stack technologies. You will engage with AV application engineers, optimize GPU compute, and contribute to library development and performance tuning.

You will collaborate across architecture, research, libraries, tools, and system software teams to push next-generation hardware and

Qualifications

• 8+ years of hands-on validated ML/DL performance engineering experience with focus on improving GPU compute efficiency of training and inferencing workloads.
• Experience with C, C++, or Python and proficiency with Linux.
• Solid understanding of software development, programming techniques, and algorithms.
• Strong mathematical fundamentals, including linear algebra and numerical methods.
• Background in parallel programming and high-performance computing, with extensive knowledge of parallel architectures and methods for performance analysis and tuning.
• Experience in GPU programming is desirable.
• Experience in distributed communication optimization is highly helpful.

Responsibilities

• Engaging directly with key AV application engineers to understand current and future problems they are solving.
• Develop and improve fundamental parallel algorithms and data structures.
• Provide efficient solutions using GPUs through library development and direct application contributions.
• Collaborate closely with the architecture, research, libraries, tools, and system software teams at NVIDIA to influence the build of next-generation architectures, software platforms, and programming models.
• Engage in deep optimization of high-performance operators, including GPU kernel optimization, instruction-level tuning, and compiler optimization.
• Support customers to use NVIDIA libraries such as cuDNN, cuBLAS, and CUTLASS and open-source libs like DeepGEMM, FlashMLA, FlashAttention, Flashinfer, etc.
• Improve communication for Physical AI-related distributed transformer workloads by employing communication tools developed by NVIDIA.

Skills

ML/DL perf eng

C/C++/Python

Linux

Parallel programming

GPU programming

Communication skills

Education

BSc/MSc/PhD or equivalent

Tools

Nsight System

Nsight Compute

CUDA kernels

NCCL

Job description

NVIDIA Corporation in Santa Clara, California, is seeking a Solutions Architect with a performance engineering background to accelerate Physical AI workloads in Autonomous Vehicles and Robotics using NVIDIA's full-stack technologies. You will engage with AV application engineers, optimize GPU compute, and contribute to library development and performance tuning.

You will collaborate across architecture, research, libraries, tools, and system software teams to push next-generation hardware and