Compiler Engineering Intern/Co-Op (Graduate | Summer 2025 | Hybrid)

WHAT YOU DO AT AMD CHANGES EVERYTHING

We care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences - the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our mission is the AMD culture. We push the limits of innovation to solve the world's most important challenges. We strive for execution excellence while being direct, humble, collaborative, and inclusive of diverse perspectives.

AMD together we advance_

An exciting internship opportunity to make an immediate contribution to AMD's next generation of technology innovations awaits you! We have a multifaceted, high-energy work environment filled with a diverse group of employees, and we provide outstanding opportunities for developing your career. During your internship, our programs provide the opportunity to collaborate with AMD leaders, receive one-on-one mentorship, attend amazing networking events, and much more. Being part of AMD means receiving hands-on experience that will give you a competitive edge. Together We Advance your career!

JOB DETAILS:

• Location: San Jose, CA, US
• Onsite/Hybrid: This role requires the student to work full time (40 hours a week), either in a hybrid or onsite work structure throughout the duration of the co-op/intern term.
• Duration: May 27, 2025, to August 8, 2025 or June 23, 2025 to September 12, 2025

WHAT YOU'LL BE DOING:

IREE is an open-source compiler toolchain that is built to be a full-stack compiler for ML models built using MLIR. It is built from the ground up to be able to target multiple hardware backends (like CPU, GPU and NPU devices). With an internship on this team, you would have the opportunity to work at any level of the compiler stack, starting from front-end like PyTorch (through Torch-MLIR), full program optimizations, or low-level code generation optimizations on CPU or GPU. You will also work on a production compiler that is targeting efficient compilation of the latest ML models (like SDXL and LLama family of models) on the latest AMD GPUs. In addition, you get to work with a team of highly motivated and dedicated engineers. Many of them are contributors/maintainers of the Structured op based compilation framework built in MLIR encompassing dialects like Linalg and Vector dialects and transformations in those dialects.

In this role you will -

• Come up to speed on IREEs compilation flow and dialects/transformations in MLIR that IREE relies upon
• Work with your mentor to -

• Scope out a specific problem that could be addressed in the internship time frame
• Build a working prototype/solution for the problem
• Ensure that there is adequate testing and maintain high code quality to be able to merge the changes into IREE/MLIR
• It is highly encouraged to make the work be part of publication in academic journals or submitted to technical conferences

WHO WE ARE LOOKING FOR:

• You are currently enrolled in a US based University in a master's or PhD program majoring in Computer Science, Computer Engineering or a related field.

• If you have knowledge/experience with any of the following technical skills (or related areas) and are enthusiastic about this role, we strongly encourage you to apply -

• Strong background in compilers and optimization techniques
• Polyhedral compilation background is a plus, but not required
• Hands on experience in LLVM/MLIR is an asset, but not necessary
• Strong Program skills in C++.
• Familiarity with Python is highly desirable

Note: By submitting your application, you are indicating your interest in AMD intern positions. We are recruiting for multiple positions, and if your experience aligns with any of intern opportunities, a recruiter will contact you.

At AMD, your base pay is one part of your total rewards package. Your base pay will depend on where your skills, qualifications, experience, and location fit into the hiring range for the position. You may be eligible for incentives based upon your role such as either an annual bonus or sales incentive. Many AMD employees have the opportunity to own shares of AMD stock, as well as a discount when purchasing AMD stock if voluntarily participating in AMD's Employee Stock Purchase Plan. You'll also be eligible for competitive benefits described in more detail here.

AMD does not accept unsolicited resumes from headhunters, recruitment agencies, or fee-based recruitment services. AMD and its subsidiaries are equal opportunity, inclusive employers and will consider all applicants without regard to age, ancestry, color, marital status, medical condition, mental or physical disability, national origin, race, religion, political and/or third-party affiliation, sex, pregnancy, sexual orientation, gender identity, military or veteran status, or any other characteristic protected by law. We encourage applications from all qualified candidates and will accommodate applicants' needs under the respective laws throughout all stages of the recruitment and selection process.