Advanced Micro Devices, Inc

Fellow, AI Workload Optimization

Advanced Micro Devices, Inc • $150K — $200K *
Bellevue, WA 98006In-Person
Information Technology
11 - 15 years of experience
Today
Be an Early Applicant
Job Overview by Ladders

Qualifications

  • 15+ years of software development experience, including 5 years in a technical leadership role.
  • Deep expertise in AI frameworks like PyTorch, JAX, and the ROCm software stack.
  • Proven track record of optimizing distributed inference and training across multi-node environments.
  • Mastery of performance profiling tools such as TorchProfiler and ROCm Profiler.
  • Strong understanding of modern model architectures and advanced optimization techniques.
  • PhD or Master's degree in Computer Science or related fields, or equivalent experience.

Responsibilities

  • Define and drive the workload optimization strategy for the AI software stack.
  • Lead profiling and tuning of large-scale AI models for optimal performance on AMD hardware.
  • Engage with top customers to understand their workload needs and deliver tailored solutions.
  • Collaborate with hardware and software teams to influence future silicon features for AI workloads.
  • Develop advanced tools for performance estimation and reporting.
  • Act as a technical ambassador and mentor within the industry and AMD's community.

Benefits

  • Comprehensive benefits package.
  • Opportunities for mentorship and career development.
  • Hybrid working model offering flexibility.
Full Job Description
Fellow, AI Software (Workload Optimization)

We are looking for a visionary technical leader to join the AI Software group. As a Fellow, you will be accountable for defining and driving the end-to-end software optimization strategy to achieve industry-leading performance for our top-tier customers. You will sit at the intersection of architecture, customer engagement, and software engineering, ensuring that AMD's software stack-from ROCm and compilers to high-level AI frameworks-is tuned to extract maximum performance for the world's most demanding AI workloads.

THE PERSON

The ideal candidate is a technical powerhouse with a proven track record of solving complex performance bottlenecks at scale. You possess deep knowledge of AI hardware architecture and software optimization, with the ability to map emerging model architectures to low-level software. You are an exceptional communicator who can translate complex technical challenges into strategic roadmaps and influence senior leadership, while also engaging deeply with key customers to solve their most critical performance needs.

KEY 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.

PREFERRED EXPERIENCE
  • 15+ years of software development experience with at least 5 years 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.

ACADEMIC CREDENTIALS
  • PhD or Master's degree in Computer Science, Electrical Engineering, or a related field, or equivalent experience.
  • Demonstrated research or applied experience in AI/ML, including areas such as deep learning, model training/inference optimization, large language models, or computer vision.


#LI-G11

#LI-HYBRID

Benefits offered are described: AMD benefits at a glance.

About Advanced Micro Devices, Inc

Advanced Micro Devices, Inc. Careers

Join the innovative forefront of technology with a career at Advanced Micro Devices, Inc. (AMD), a leader in semiconductor development. As part of our global team, you will contribute to an organization renowned for its dedication to innovation, leadership, and diversity in the tech industry.

Work You’ll Do

At AMD, we offer job opportunities that push the boundaries of what is possible. Our team is composed of professionals who lead the way in microprocessor and graphics technology, driving industry standards and innovation. With AMD, you will be part of a culture that values growth and professional development, ensuring that every team member has the opportunity to excel.

Transform Your Career

AMD is not just about advancing technology, but also about advancing careers. Whether you are looking for an internship, a full-time position, or leadership roles, AMD provides the platform to propel your career to new heights. Our commitment to professional growth is matched by our dedication to diversity and inclusion, making AMD a place where everyone can thrive.

Innovative Work Environment

Join a team of over 12,000 dedicated professionals at the intersection of technology, industry expertise, and digital innovation. At AMD, you will work on groundbreaking projects that shape the future of computing and graphics. Our collaborative environment encourages networking and the sharing of ideas across teams and disciplines.

Career Development and Benefits

AMD is committed to the development of its employees. We offer robust training programs, including leadership development and diversity training, to ensure our team is equipped for both current challenges and future opportunities. Our benefits package is designed to support the well-being and financial security of our employees and their families.

Explore Job Opportunities

From engineering to marketing, AMD offers a range of career paths that cater to diverse skills and interests. Our hiring process is designed to be transparent and engaging, helping you to understand where you fit within our team and how you can contribute to our collective goals.

Stay Connected

Join Our Team Search open positions that match your skills and interest. We look for passionate, curious, creative, and solution-driven team players. Explore the opportunities to join a company that’s committed to your career growth and to innovation in the technology sector.

Keep Up to Date

Stay ahead with career tips, insider perspectives, and industry-leading insights you can put to use today—all from the people who work here.

Job Alert Emails

Personalize your subscription to receive job alerts, latest news, and insider tips tailored to your preferences. Discover the exciting and rewarding career opportunities that await at Advanced Micro Devices, Inc.

Interview and Resume Tips

Prepare for your future with AMD by accessing resources that help you craft your resume and excel in interviews. Our goal is to help you showcase your best professional self and align your skills with the needs of our dynamic team. At Advanced Micro Devices, Inc., we empower our employees to innovate, lead, and grow. Join us in driving the future of technology while building a rewarding and sustainable career.
Learn more about Advanced Micro Devices, Inc
Size
15,500 employees
Market Cap
$100.9 billion
Industry
Manufacturing & Automotive
Net Income
$2.4 billion
Founded
1969
5 Year Trend
+30.9%
Revenue
$9.7 billion
NASDAQ
AMD
* Ladders Estimates

Similar Jobs

Ad banner background

Get Ready For Your
Next Interview

More Jobs at Advanced Micro Devices, Inc

More Information Technology Jobs

Find similar Fellow, AI Workload Optimization jobs:

NationwideBellevue, WA