Masters AI Training Software & Performance Co-op

ADVANCE YOUR CAREER. ADVANCE THE WORLD. 

At AMD, we believe technology has the power to solve the world’s most important challenges. From advancing healthcare and scientific discovery to powering AI and the technologies people rely on every day, innovation at AMD is shaping the future. 

Whether you’re designing next-gen processors, enabling AI breakthroughs, or bringing leading edge products to market, every role at AMD contributes to something bigger — technology that moves the world forward. Join us and, together, we’ll advance your career.

As an AMD co-op, you’ll be placed at the epicenter of the AI ecosystem, working alongside experts and industry pioneers. You’ll do important work, learn new skills, expand your network, and gain real-world experience on projects that impact millions of end-users worldwide. Whether you’re an undergrad or a PhD student, your contributions matter—and your experience here will be a launchpad for what comes next.

JOB DETAILS:

• Location: San Jose, CA, USA
• Onsite: This role requires the student to work full time (40 hours a week), onsite work structure throughout the duration of the co-op/intern term. 
• Duration: Spring/Summer 2027 Co-Op: January 25, 2027 - August 13, 2027

WHAT YOU WILL BE DOING:

Would you like to be part of a world-class team enabling software for world-class datacenters and the mightiest supercomputers? AMD is searching for talented and highly motivated AI Software Engineers to join our team of developers pushing the boundaries of efficiency and performance to enable and optimize the software ecosystem for the next generation of GPU computational accelerators.

Our team has an unparalleled perspective of the AI landscape. We work with the industry's most sophisticated clients to help them leverage the latest hardware capabilities for their AI use cases. As part of our team, you will be among the first in the world to combine the newest hardware with the industry's latest applications, libraries, frameworks, and SDKs to push the limits of innovation and solve the world's most complex challenges.

In this role, you will:

• Work with a team of software engineers and architects developing AMD Instinct™ GPU AI software.
• Develop tools for debugging distributed training at a large scale.
• Analyze and optimize the performance of AI software workloads across training and fine-tuning.
• Identify hardware and software bottlenecks across CPUs, GPUs, memory systems, and communication infrastructure.
• Profile and optimize performance across the AI software stack using advanced performance analysis tools.
• Investigate and develop optimization strategies to improve scalability and efficiency of AI workloads.
• Implement and optimize parallel computing methods on GPU-accelerated systems.
• Benchmark, evaluate, and tune AI frameworks, libraries, SDKs, and applications on AMD platforms.
• Collaborate with architects and engineering teams to drive next-generation AI software performance solutions.

WHO WE ARE LOOKING FOR:

• Currently enrolled in a master's program in Computer Science, Computer Engineering, Artificial Intelligence, Machine Learning, or a related field.
• Strong programming skills in C++ and/or Python with Agentic coding experience.
• Strong development experience with at least one major deep learning framework such as PyTorch, JAX, TensorFlow, VLLM, or SGLang
• Expertise in profiling tools across the AI software stack including PyTorch Profiler, ROCm Profiler, VTune, and Nsight.
• Experience implementing and optimizing parallel methods on GPU accelerators (NCCL/RCCL, OpenMP, MPI).
• Advanced performance analysis skills for both CPU and GPU architectures.
• Deep understanding of latest LLM models and its architectures.
• Deep understanding of various parallelisms (DP/TP/EP/CP) and their corresponding collectives. 
• Understanding of AI software optimization techniques and performance tuning methodologies.
• Publications in peer-reviewed AI, Machine Learning, High Performance Computing, or Computer Architecture conferences and journals are a plus.

This role is not eligible for visa sponsorship.

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.

AMD may use Artificial Intelligence to help screen, assess or select applicants for this position.  AMD’s “Responsible AI Policy” is available here.

This posting is for an existing vacancy.