Senior GPU Inference Performance Engineer
- Discuss this with your agent
- Open in Claude
- Open in ChatGPT
Role details
Tech stack
+8 more
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.comGood distractions
Talks and stories from around this role ā technically off-topic, practically not.
Moments
Explore playlistsVideos
See allRelated articles
See all
Dev Digest 157: CUDA in Python, Gemini Code Assist and Back-dooring LLMs
Got AI ideas but no money? Here are 10 free ways to level up your AI skills with Google Cloud
Stephan Gillich - Bringing AI Everywhere
Dev Digest 120 - Apple and peers