The role is for a systems software engineer focused on ensuring AI stack readiness for NVIDIA's DGX Station. The engineer will work on optimizing AI applications and deep learning frameworks to deliver high performance on multi-GPU architectures. This position involves profiling workloads, identifying bottlenecks, and collaborating with various teams to enhance performance. The engineer will also validate the full NVIDIA AI software stack and support customer deployment readiness.

• BS or MS or equivalent experience in Computer Science, Electrical Engineering, or related field.
• 12+ years in systems software engineering with hands-on experience in AI/ML workload optimization.
• Strong proficiency with deep learning frameworks such as PyTorch, TensorFlow, or JAX.
• Experience profiling and optimizing GPU workloads using tools like Nsight Systems and Nsight Compute.
• Strong understanding of GPU architecture and its impact on AI workload performance.
• Proficiency in C/C++, CUDA, and Python.

• Own production readiness of AI applications on DGX Station.
• Work cross-functionally to profile and optimize deep learning workloads across training and inference.
• Identify bottlenecks in GPU compute and implement optimizations across the stack.
• Collaborate with framework and compiler teams to improve kernel fusion and memory management.
• Validate multi-user and concurrent workload scenarios to ensure reliable performance.
• Validate the full NVIDIA AI software stack for version compatibility and performance parity.
• Build and maintain performance benchmarking infrastructure for automated regression tracking.
• Work with product management and partners to understand target use cases and ensure performance.

• Employees at NVIDIA are often offered comprehensive, day-one benefits—including medical, dental, and vision coverage with HSA support, life and disability insurance, an Employee Assistance Program, and a 401(k) with auto-enrollment. Many roles also have generous time off and holidays, donation matching (up to $10,000), and a wide menu of extras like FSAs, commuter benefits, legal and identity-theft protection, pet insurance, and wellness discounts. Optional programs can include student-loan and home-purchase support, plus family care resources and expert medical services.