Senior Forward Deployed Engineer; FDE) AI and GenAI Inference Optimization
Listed on 2026-10-23
-
Software Development
AI Engineer (Applied/Software), Machine Learning/ ML Engineer
We are seeking a Senior Member of Technical Staff to join our Forward Deployed Engineering (FDE) team. In this customer-facing role, you will work directly with strategic partners and customers to enable, optimize, and deploy state-of-the-art GenAI inference workloads on AMD data center CPUs and GPUs (Instinct™ family). You will bridge the gap between cutting‑edge AI models and AMD hardware, driving competitive performance across the full inference stack.
Responsibilities- Partner directly with customers to enable and optimize large language models (LLMs) and multimodal models on AMD Instinct GPUs, meeting committed performance and accuracy targets.
- Optimize GenAI inference pipelines on vLLM and SGLang, including PD disaggregation, expert parallelism (EP/wide-EP), tensor/data parallelism, speculative decoding, and KV cache management.
- Develop and tune low-level GPU kernels (Triton, HIP, composable kernels) for attention, MoE, GEMM, quantization, and normalization operations.
- Profile end-to-end workloads, identify bottlenecks across GPU/CPU boundaries, and deliver measurable throughput and latency improvements versus competitive targets.
- Enable advanced data types and techniques (FP8, MXFP4, sparse attention) with a focus on both performance and numerical accuracy.
- Debug complex runtime, stability, and accuracy issues in production environments; partner with core engineering teams to identify root causes and contribute fixes.
- Contribute optimizations back to open-source frameworks (vLLM, SGLang, ATOM, AITER) and internal libraries; document best practices and deliver production-ready artifacts to customers.
- Leverage AI-assisted coding tools and agentic workflows to accelerate development, debugging, optimization, and code review across the inference stack.
- BS, MS, or PhD in Computer Science, Electrical Engineering, or a related field, along with 8+ years of relevant experience.
- Deep expertise in GPU architecture and performance optimization, including hands‑on experience with AMD ROCm (or CUDA) and GPU kernel development.
- Strong proficiency in Python and C++, with experience developing and optimizing kernels using Triton, HIP, or CUDA.
- Experience with AI-assisted coding and agentic development tools (e.g., Git Hub Copilot, Claude, Cursor, or equivalent) to enhance productivity in kernel development, large-scale codebase navigation, debugging, and framework integration.
- Experience working with LLM inference frameworks (vLLM, SGLang, or equivalent) and serving optimization techniques.
- Solid understanding of transformer and MoE architectures, quantization, and distributed inference strategies (TP, DP, EP, and related parallelism techniques).
- Proficiency with profiling and debugging tools (e.g., roc Profiler, ROCdebug-agent).
- Experience optimizing frontier models (Deep Seek, Qwen, Kimi, Mini Max, GLM, etc.) in production environments.
- Track record of contributing to open-source AI inference frameworks.
- Experience integrating AI-driven automation into engineering workflows and developer tooling.
- Strong customer-facing communication skills and the ability to manage technical deliverables against committed timelines.
AMD is committed to building a diverse and inclusive workplace. We encourage candidates to apply even if their experience does not align perfectly with every qualification listed. If you are excited about this role and believe you can make an impact, we would welcome your application.
#LI-MH3
Benefits offered are described: AMD benefits at a glance
AMD does not accept unsolicited resumes from headhunters, recruitment agencies, or fee-based recruitment services. AMD and its…
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).