← All roles

Fellow, AI Workload Optimization

AMD· Engineering
Apply on company site
Bellevue, WA$256kFull-timePosted 4mo ago

About the role

Technical leadership role focused on defining and driving end-to-end software optimization strategy to achieve industry-leading performance for AI workloads on AMD hardware, spanning architecture, customer engagement, and software engineering across ROCm, compilers, and high-level AI frameworks.

Responsibilities

  • Strategic Leadership: Set the technical vision and roadmap for workload optimization across the AI software stack, ensuring AMD remains the platform of choice for top-tier AI customers.
  • Workload Performance Engineering: Lead the profiling, analysis, and tuning of large-scale models (LLMs, Diffusion, Multimodal, and MoE) to ensure "out-of-the-box" performance excellence on AMD hardware.
  • Customer Engagement: Partner with top customers and hyperscalers to understand their unique workload requirements and deliver tailored architectural wins and software optimizations.
  • Hardware-Software Co-design: Collaborate across hardware architecture, compiler, and framework teams to influence future silicon features based on evolving AI workload trends.
  • Ecosystem Innovation: Drive the development of advanced tools and frameworks for performance estimation, modeling, and automated reporting.
  • Community & Mentorship: Act as a technical ambassador in industry forums and open-source communities. Mentor and inspire the next generation of AMD’s technical leaders and engineers.

Qualifications

  • Deep software development experience with experience in a high-level technical leadership role (Fellow or equivalent).
  • Deep expertise in AI Frameworks (PyTorch, JAX, vLLM, SGLang) and the ROCm software stack.
  • Proven history of optimizing distributed inference and training at scale across multi-node/multi-GPU environments.
  • Mastery of performance profiling tools (e.g., TorchProfiler, ROCm Profiler, Nsight) and hardware-level performance modeling.
  • Strong understanding of modern model architectures (Transformer, Attention, KV Cache) and optimization techniques like quantization, speculative decoding, and FlashAttention.
  • Demonstrated ability to drive cross-functional initiatives in fast-paced, ambiguous environments.

Skills

Required:Strategic leadership and roadmap planningAI workload optimizationPerformance profiling tools

Benefits

  • Benefits offered are described: AMD benefits at a glance
Last updated Oct 22, 2026. Interested? Apply on the company site.
AMD logo

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
Company profileWebsite

More open roles at AMD