Deep Learning Kernel Software Performance Architect

NVIDIA

📍 Shanghai, China, China

Full-time other-general

Job Description

NVIDIA is seeking Software Performance Architects to optimize GPU kernel performance for state-of-the-art data-center platforms. We build automated, data-driven workflows to detect, explain, and prevent performance regressions across key deep learning workloads, partnering closely with kernel developers, compiler teams, infrastructure, and architecture/performance groups.


What you'll be doing:
+ Performance analysis, optimization and debugging
+ Build performance narratives using structured methodology: baselines, projections, controlled comparisons, and regression attribution.
+ With the methodologies, analyze performance of GPU-accelerated kernels and key deep learning building blocks, identify gaps with baselines or projections, then optimize the kernels' performance to fill the gaps.
+ Debug performance issues end-to-end: reproduce, isolate root causes, propose fixes or mitigation paths, and drive closure with the owning teams.
+ Automation + regressi...
Apply for this Position