Senior AI Systems Engineer - GPU & Performance Optimizer

Nvidia Corporation; Santa Clara (CA); On-site; USD 224,000 - 356,500; Full time

Job summary

Nvidia Corporation is looking for an experienced professional to optimize AI workloads at DGX Station (Galaxy) in Santa Clara, California. This role involves ensuring production readiness of AI applications, profiling multi-GPU environments, and collaborating to enhance NVIDIA's software stack.

The ideal candidate has over 12 years of experience in systems software engineering and strong expertise with deep learning frameworks. The position offers an attractive salary with equity and benefits.

Qualifications

• 12+ years in systems software engineering with hands-on experience.
• Strong proficiency with deep learning frameworks.
• Experience profiling and optimizing GPU workloads.

Responsibilities

• Define and validate production readiness of AI applications.
• Profile and optimize workloads for multi-GPU architectures.
• Collaborate to improve kernel fusion and memory management.

Skills

AI/ML workload optimization

GPU performance analysis

Deep learning frameworks (PyTorch, TensorFlow, JAX)

C/C++

CUDA

Python

Education

BS or MS in Computer Science or related field

Tools

Nsight Systems

Nsight Compute

CUPTI

Job description

Nvidia Corporation is looking for an experienced professional to optimize AI workloads at DGX Station (Galaxy) in Santa Clara, California. This role involves ensuring production readiness of AI applications, profiling multi-GPU environments, and collaborating to enhance NVIDIA's software stack.

The ideal candidate has over 12 years of experience in systems software engineering and strong expertise with deep learning frameworks. The position offers an attractive salary with equity and benefits.