AI Stack Optimization Engineer at NVIDIA

Join NVIDIA as a Senior Systems Software Engineer, specializing in AI stack performance for the DGX Station. Take charge of optimizing AI applications for high-performance, multi-GPU setups.

In this technical role, you will be responsible for ensuring AI application readiness on the DGX Station, which includes Nemo Claw and Hermes agents. Engage with various teams to profile workloads and enhance performance across different frameworks like PyTorch and Tensor Flow. You will also focus on resolving bottlenecks to deliver exceptional throughput in both single and multi-GPU environments.

Key Responsibilities:

• Own AI application readiness for production on DGX Station

• Profile and optimize LLM deep learning workloads across frameworks

• Identify system-level bottlenecks in GPU compute and NVLink

• Collaborate with compiler and kernel teams for optimization

• Validate multi-user scenarios for shared workstation reliability

Requirements:

• Proven experience in systems software engineering for AI

• Strong background in GPU performance analysis and optimization

• Familiarity with DL frameworks like PyTorch and Tensor Flow

• Experience with concurrent workload scenarios

• Technical expertise in memory management and scheduling

Leverage your expertise to drive performance for NVIDIA's cutting-edge AI solutions on DGX Station.
#J-18808-Ljbffr