Senior GPU Inference Performance Engineer

Advanced Micro Devices, Inc.
Santa Clara, CA, United States
about 1 month ago

Role details

Contract type
Permanent contract
Employment type
Full-time (> 32 hours)
Experience level
Expert
Working hours
Regular working hours

Tech stack

Artificial Intelligence Nvidia CUDA Computer Networks Computer Engineering Extract Transform Load (ETL) Memory Management Firmware Python (Programming Language) Open Source Technology PCI Express Remote Direct Memory Access Software Engineering
+8 more
Graphics Processing Unit (GPU) Performance Testing Large Language Models Parallel Computation Kubernetes Information Technology Decoding Docker

Job description

We are looking for a Senior GPU Inference Performance Engineer to own end-to-end performance analysis of GPU-accelerated AI inference workloads. You will profile, diagnose, and explain performance across the full stack, from GPU silicon through the software runtime, and drive competitive positioning against other accelerator vendors. This role sits at the intersection of hardware, systems software, and AI serving frameworks, and requires someone who can go deep on a trace and present findings to product and executive stakeholders., * Full-stack GPU profiling: Instrument and analyze inference workloads across AMD Instinct (ROCm, rocProfiler, Omniperf) and NVIDIA (CUDA, Nsight Systems/Compute, DCGM) GPUs. Identify bottlenecks spanning HBM bandwidth, compute utilization, kernel scheduling, memory allocation, and PCIe/Infinity Fabric data movement.

  • AI serving framework performance: Profile and optimize inference engines including vLLMSGLang, and emerging serving runtimes. Understand KV-cache management, continuous batching, PagedAttention, speculative decoding, and quantization (FP8, MXFP4, INT4) effects on throughput and latency.
  • Competitive performance analysis: Design and execute head-to-head benchmarks (AMD vs. NVIDIA) on standardized LLM workloads. Produce clear, data-backed explanations of why performance differs - attributing gaps to specific hardware features (HBM bandwidth, compute density, interconnect topology), software maturity (kernel libraries, operator fusion, graph compilation), or configuration differences.
  • Multi-server inference networking: Profile and optimize distributed inference topologies including prefill-decode (PD) disaggregationpipeline parallelism, and tensor parallelism across multi-node clusters. Analyze network-level bottlenecks using RDMA/RoCE traces, NCCL/RCCL collective profiling, and NIC-level counters (Pensando, ConnectX). Quantify the impact of network latency, bandwidth, and congestion on end-to-end inference SLAs.
  • GPU operator and Kubernetes stack: Profile the overhead introduced by GPU operators, device plugins, container runtimes (Docker, containerd), and Kubernetes scheduling on inference latency. Identify and resolve jitter, cold-start, and resource contention issues in production serving environments.
  • Tooling and automation: Build reproducible benchmarking harnesses, profiling scripts, and performance regression dashboards. Automate trace collection and analysis to support continuous performance validation across driver, firmware, and framework updates., AMD may use Artificial Intelligence to help screen, assess or select applicants for this position. AMD’s ā€œResponsible AI Policyā€ is available here.

Requirements

A hands-on performance engineer who is equally comfortable reading a GPU trace and briefing executives. You are curious, evidence-driven, rigorous and you don’t stop at ā€œX is faster,ā€ you explain why, rooted in hardware and software evidence. You collaborate across hardware, systems software, and AI serving framework teams, communicate clearly in written reports and presentations, and thrive at the intersection of silicon, systems, and AI., * Background in GPU performance engineering, HPC, or systems performance analysis

  • Hands-on proficiency with either AMD (ROCm, rocProfiler, Omniperf/Omnitrace) or NVIDIA (CUDA, Nsight Systems/Compute, NCU) profiling toolchains, with deep understanding of GPU architecture: warp/wavefront execution, memory hierarchy (registers ? LDS/shared ? L2 ? HBM), occupancy, and instruction-level parallelism
  • Experience profiling vLLM, SGLang, or equivalent LLM serving frameworks, including quantization workflows (FP8, MXFP4, INT4, AWQ, GPTQ) and their performance implications
  • Experience with multi-GPU and multi-node inference - tensor parallelism, pipeline parallelism, or PD disaggregation over RDMA/RoCE - including RCCL/NCCL profiling and network tools (perftest, ib_write_bw, tcpdump, Memory Fabric counters)
  • Demonstrated ability to explain performance differences in written reports or presentations - not just ā€œX is fasterā€ but why, rooted in hardware and software evidence
  • Strong Python and C/C++ skills; comfort reading GPU kernel code (HIP/CUDA)
  • Experience with Kubernetes GPU scheduling, MIG, and GPU operator performance, or contributions to open-source inference or profiling projects

ACADEMIC CREDENTIALS:

  • Bachelor’s degree in Computer Science, Computer Engineering, Electrical Engineering, or a related technical field preferred; advanced degree desired

About the company

At AMD, our mission is to build great products that accelerate next-generation computing experiences-from AI and data centers, to PCs,

gaming

and embedded systems. Grounded in a culture of innovation and collaboration, we believe real progress comes from bold ideas, human

ingenuity

and a shared passion to create something extraordinary. When you join AMD,

you’ll

discover the real differentiator is our culture. We push the limits of innovation to solve the world’s most important challenges-striving for execution excellence, while being direct, humble, collaborative, and inclusive of diverse perspectives. Join us as we shape the future of AI and beyond.

Together, we advance your career.

Apply for this position

This job is hosted externally. Click below to view the full posting and apply.

Apply on www.techcareers.com

Good distractions

Talks and stories from around this role — technically off-topic, practically not.

2:07 min

Inspecting default bridge architectures and custom Docker networks

Oliver Seitz Oliver Seitz Ā· WWC 2025

2:03 min

Accelerating token generation speeds with speculative decoding techniques

Christin Pohl Christin Pohl Ā· WWC Europe 2026

2:19 min

Orchestrating over-the-air firmware updates for vehicle modules

Denis Grahovac Ā· WWC 2021

1:34 min

Profiling and debugging GPU code with Nsight developer tools

Paul Graham Paul Graham Ā· WWC 2025

2:34 min

Docker sandbox architecture and microVM environment integration

Manuel de la Peña Manuel de la Peña · WWC Europe 2026

2:32 min

Core libraries driving inference engines and multi-GPU networking

Adolf Hohl Adolf Hohl Ā· WWC 2024

Videos

See all

Related articles

See all