Principal GenAI Inference Optimization Engineer
San Jose, CA$240k - $360kFull-timePosted 6mo ago
About the role
Principal GenAI Inference Optimization Engineer at AMD focusing on improving performance, efficiency, and scalability of generative AI inference workloads on AMD GPU platforms across single-node and distributed environments.
Responsibilities
- Optimize performance of GenAI inference workloads on AMD GPU platforms across single-node and distributed environments.
- Improve latency, throughput, and cost efficiency for LLM and multimodal model serving in production.
- Analyze and resolve bottlenecks across compute, memory, and communication (e.g., kernel efficiency, KV-cache usage, memory bandwidth, scheduling).
- Contribute to cross-stack optimizations spanning kernels, runtimes, communication libraries, and inference/serving frameworks (e.g., vLLM, SGLang, Triton, or similar systems).
- Implement and evaluate inference optimization techniques such as batching strategies, quantization, prefix caching, and speculative decoding.
- Support development and optimization of scalable serving systems, including request scheduling and resource utilization.
- Develop and use profiling, benchmarking, and performance analysis tools for inference workloads.
- Collaborate with hardware, compiler, and framework teams to improve overall system performance.
- Contribute to internal tools and, where applicable, open-source projects for inference optimization on AMD platforms.
- Document best practices and contribute to performance guidelines for GenAI deployment.
Qualifications
- Strong understanding of GPU architecture and performance fundamentals (compute, memory hierarchy, interconnects such as PCIe/Infinity Fabric/RDMA).
- Experience with GenAI inference optimization techniques (e.g., quantization, KV-cache optimization, batching).
- Hands-on experience with inference/serving frameworks such as vLLM, SGLang, Triton, TensorRT-LLM, or similar.
- Experience working on LLM or multimodal inference workloads.
- Familiarity with distributed systems and serving architectures.
- Experience with ML frameworks (PyTorch, JAX, or TensorFlow), especially for inference.
- Proficiency in Python and at least one systems language (C++/CUDA/HIP).
- Experience with profiling, debugging, and performance tuning tools.
- Ability to work collaboratively across teams and deliver impactful optimizations.
Skills
Required:GPU architecture and performance fundamentalsGenAI inference optimization techniquesinference/serving frameworksLLM/multimodal workloadsdistributed systemsML frameworksPythonsystems programming
Benefits
- Benefits offered are described: AMD benefits at a glance.
Last updated Oct 15, 2026. Interested? Apply on the company site.
About AMD
together we advance_AI
AMD delivers leadership high-performance and adaptive computing solutions to advance data center AI, AI PCs, intelligent edge devices, gaming, & beyond.
AIData CenterHigh-Performance ComputingGamingEdge Devices
More open roles at AMD
- System Thermal Design Staff Engineer4d agoPenang, Malaysia · Engineering
- PD Engineer4d agoHybrid (Bangalore) · Engineering
- PD Engineer4d agoBangalore · Engineering
- Low-Power & CDC Verification Engineer4d agoSanta Clara, California · Engineering
- Technical Program Manager4d agoShanghai · Engineering