Jobs Companies NVIDIA Deep Learning Kernel Software Performance Architect

About this Deep Learning Kernel Software Performance Architect role at NVIDIA

NVIDIA · Onsite · China, Shanghai

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 + regression infrastructure (Python-heavy)

    • Develop and maintain Python-based automation for performance testing and analysis—using modern AI-assisted developer tools (e.g., Cursor/Claude Code/Copilot) to accelerate scripting while keeping code maintainable and reviewable.

    • Design and operate performance test workflows: coverage definition, test/workload generation, automated large-scale execution (CI/nightly/on-demand), rerun rules, and reproducibility standards.

  •  Cross-team collaboration and operating model

    • Work with kernel developers and the compiler teams to ensure performance checks are practical, scalable, and aligned to release needs.

    • Work with chip architecture and modeling teams to solidify the performance methodology across chip architecture generations and common Deep Learning operators such as GEMM, Attention, MoE.

    • Partner with SWQA and infrastructure teams for execution at scale and reliable pipelines/dashboards.

  • Following general software engineering best practices including support for regression testing and CI/CD flows

What we need to see:

  • Masters or PhD degree or equivalent experience in Computer Science, Computer Engineering, Applied Math, or related field

  • Strong programming ability in Python plus C/C++ with 2+ working experience (performance-oriented code reading/debugging)

  • Solid fundamentals in computer architecture, parallel programming and performance reasoning (latency/throughput, memory hierarchy, parallelism) to be able to identify bottlenecks, optimize resource utilization, and improve throughput

  • Experience with performance analysis workflows: profiling, measurement methodology, reproducibility, and regression triage.

  • Comfortable working across teams and driving issues to decision/closure with clear communication

Ways to stand out from the crowd:

  • Experience with high-performance kernels or math libraries (e.g., GEMM/attention, CUTLASS-like concepts)

  • GPU programming/perf experience (CUDA or equivalent parallel programming)

  • Strong ML/DL workload understanding (training/inference shapes, precision modes, perf bottlenecks)

  • Familiarity with simulators/analytical modeling or performance characterization methodology

Ready to apply to NVIDIA?
Apply to NVIDIA

About NVIDIA

NVIDIA pioneered accelerated computing. Today, our AI infrastructure powers global intelligence, transforming every industry. Learn more about NVIDIA .

See all jobs at NVIDIA →

Similar jobs

Sign up for suggestions tailored to the jobs you open and the searches you save.

More jobs at NVIDIA

See all jobs at NVIDIA →

Apply now
🤖

Whoa — hold up

JobsRadar was built for real people having a rough time in their job search — not for automated requests. You're clicking way too fast and you're now temporarily blocked.

Come back later. If you're genuinely job hunting, we've got your back — just act like a human.

Catch your next role the second it’s posted.

Create a free account and we’ll watch the boards for you — the instant a job matches your search, it lands in your inbox or Telegram. No digging, no refreshing.

Create free account

Free forever · takes 30 seconds · already have one?

Get an edge on your job hunt.

Join our Telegram channel for the stuff that helps you land the role — salary benchmarks, the weekly market pulse, and new-feature drops. No spam, just signal.

Join the channel — it's free