算子开发实习生 AI/ML Kernel Optimization Intern (Jul - Dec 2025) Job Listing at AMD in Shanghai (Job ID 63322-en-us)

Description

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_

AI/ML Kernel Optimization Intern

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: Shanghai, China

• Onsite/Hybrid: This role require the student to work full time (40 hours a week), either in a hybrid (minimum 3 Days in Office) or onsite work structure throughout the duration of the co-op/intern term.  

• Duration: July 1, 2025 to Dec 31, 2025

WHAT YOU WILL BE DOING:

We are seeking a highly motivated Machine Learning (ML)/Artificial Intelligence (AI) intern/co-op to join our team and contribute to the development of next-generation product differentiation features alongside expert ML/AI engineers. In this role, you will: 

• Gain hands-on experience with cutting-edge technologies in ML, AI, and High-Performance Computing. 

• Learn to analyze and optimize GPU Kernel to maximize performance for specific AI operations.

• Contribute to projects such as: 

• Researching, developing, and deploying machine learning and computer vision solutions for AMD's current and future products. 
• Work closely with internal teams to analyze and improve training and inference performance on AMD GPUs.

• Design and optimize deep learning models specifically for AMD GPU performance. 

• Assisting AI software teams with roadmap planning, collateral development, and customer engagements. 
• Engage with framework maintainers to ensure code changes are aligned with requirements and integrated upstream.
• Apply sound engineering principles to ensure robust, maintainable solutions.

WHO WE ARE LOOKING FOR: 

• You are currently enrolled in a China based University in a Master's program 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:   

• Machine Learning & Data Science: Exposure to machine learning algorithms, data analysis, computer vision, etc. through coursework or projects. 

• Programming Languages: Strong programming skills in C++ with a focus on writing clean, efficient, and scalable code. 

• Machine Learning Frameworks: Practical experience with machine learning libraries or frameworks such as TensorFlow, PyTorch, XLA, and JAX are preferred. 
• GPU Programming: Experience with GPU programming frameworks such as CUDA or HIP for accelerating machine learning workloads is a plus.
• High-Performance Computing: Experience in running large-scale workloads on heterogeneous compute clusters, optimizing for efficiency and scalability. 
• Compiler Optimization: Solid understanding of compiler theory and tools like LLVM and ROCm for kernel and system performance optimization. 

ACADEMIC CREDENTIALS:  

• Currently enrolled in a Master's Degree in Computer Science, Computer Engineering, Electrical Engineering, or a related field. 

#LI-EJ1

#LI-HYBRID

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