Advanced Micro Devices, Inc

Principal Software Development Engineer - AI/ML

Advanced Micro Devices, Inc • $160K — $190K *
San Jose, CA 95123In-Person
Information Technology
Less than 5 years of experience
Today
Be an Early Applicant
Job Overview by Ladders

Qualifications

  • 5-7 years of experience in compiler technology and performance optimization.
  • Proficient in C++ and object-oriented software development.
  • Hands-on experience with GPU programming frameworks like HIP or CUDA.
  • Deep understanding of compiler infrastructures such as LLVM and MLIR.
  • Experience in profiling and optimizing AI workloads and performance-critical systems.
  • Familiarity with machine learning frameworks and their execution pipelines.
  • Bachelor's or Master's degree in a relevant technical field.

Responsibilities

  • Collaborate with architecture specialists to enhance AMD products.
  • Design, develop, and optimize compiler transformations in MLIR and LLVM.
  • Create scalable solutions for AMD GPU architecture performance improvements.
  • Work with teams on advanced optimization strategies for AI workloads.
  • Implement and maintain ONNX operators and related microbenchmark suites.
  • Lead optimization efforts for various AI training and inference workloads.
  • Analyze execution characteristics to improve performance and efficiency.

Benefits

  • Comprehensive medical, dental, and vision coverage.
  • 401(k) plan with company match.
  • Generous paid time off policy.
  • Support for ongoing education and professional development.
  • Access to employee wellness programs.
Full Job Description
THE ROLE:

We are seeking a highly motivated and technically exceptional PMTS AI/ML Compiler Engineer to join AMD's AI Software organization. In this role, you will drive compiler and performance optimizations that enable state-of-the-art AI training and inference workloads on AMD GPU platforms. You will work at the intersection of compiler technology, machine learning frameworks, GPU architecture, and performance engineering, contributing to the ROCm software ecosystem and ONNX Runtime execution stack. The ideal candidate combines deep compiler expertise with hands-on experience optimizing AI workloads, and thrives in a fast-paced environment collaborating across architecture, kernel, runtime, and framework teams.

THE PERSON:

You are passionate about building high-performance AI software and solving complex optimization challenges. You possess strong compiler fundamentals, GPU programming expertise, and a proven ability to improve performance for large-scale AI models.

You enjoy collaborating with architecture, compiler, runtime, and machine learning teams to deliver innovative solutions that scale across current and future AMD GPU platforms. You have a strong sense of ownership, excellent problem-solving skills, and a track record of driving meaningful performance improvements across software stacks.

KEY RESPONSIBILITIES:
  • Work with AMD's architecture specialists to improve future products
  • Design, develop, and optimize compiler transformations and operator lowerings within MLIR, LLVM, and the ROCm software stack.
  • Develop scalable solutions that improve portability and performance across AMD GPU architectures.
  • Collaborate with compiler and runtime teams to develop advanced optimization strategies for AI workloads.
  • Implement, validate, and maintain ONNX operators and microbenchmark test suites.
  • Lead performance optimization efforts for AI training, fine-tuning, reinforcement learning (RL), and inference workloads on AMD GPUs.
  • Design and implement graph-level, kernel-level, and runtime-level optimizations that deliver measurable performance gains.
  • Analyze workload execution characteristics and identify opportunities for improvements in compute efficiency, memory utilization, and communication overhead.
  • Drive performance characterization and benchmarking across pre-silicon and post-silicon environments.


PREFERRED EXPERIENCE:
  • Strong proficiency in C++ and object-oriented software development.
  • Hands-on experience with GPU programming technologies such as HIP, CUDA, Triton, or equivalent parallel programming frameworks.
  • Experience with compiler technologies including MLIR, LLVM, or related compiler infrastructures.
  • Strong understanding of machine learning frameworks, model execution pipelines, and AI workload optimization.
  • Experience profiling and optimizing performance-critical software systems.
  • Proficiency with Linux development environments, source control systems (Git), debugging tools, and performance profilers.
  • Strong analytical, communication, and problem-solving skills.


ACADEMIC CREDENTIALS:
  • Bachelor's or Master's degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent


This role is not eligible for visa sponsorship.

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 Principal Software Development Engineer - AI/ML jobs:

NationwideSan Jose, CA