Spring/Summer 2027 PhD AI Agentic/ML System Co-op

Advanced Micro Devices, Inc.; San Jose (CA); On-site; USD 48,000 - 62,000; Full time

Job summary

Advanced Micro Devices, Inc. (AMD) invites a highly motivated ML co-op to join in San Jose for Spring/Summer 2027. You will collaborate with research and engineering teams to optimize open-weight LLMs and accelerate agentic AI on modern hardware.

Applicants should be enrolled in a master’s or PhD program in CS or EE, be proficient in Python and/or C++, and have exposure to PyTorch or JAX with a curiosity-driven approach to new techniques in a fast-paced environment.

Qualifications

• Enrolled in a master’s or PhD program in CS, EE, or related field at a US university.
• Proficient in Python and/or C++, with exposure to performance profiling or optimization.
• Familiar with ML software stacks and models (e.g., PyTorch or JAX).
• Strong motivation, curiosity and ability to learn quickly.
• Experience in agentic software engineering or ML compiler topics is a plus.

Responsibilities

• Collaborate with research and engineering teams on agentic ML system design.
• Explore techniques to improve efficiency, scalability and performance of open-weight LLMs on AMD GPUs.
• Contribute to software/hardware optimizations for ML workloads across AMD GPUs and frameworks.
• Document findings and contribute to reports, presentations, and potential publications.

Skills

Python

C++

Profiling

Optimization

Education

Master's or PhD student

Tools

PyTorch

JAX

Job description

ADVANCE YOUR CAREER. ADVANCE THE WORLD.

At AMD, we believetechnology 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 AMDis shapingthefuture.

Whetheryou’redesigning next-gen processors, enabling AI breakthroughs, orbringing leading edge products to market, every role at AMD contributes to something bigger— technologythat moves the world forward.Join us and, together, we’ll advance your career.

JOB DETAILS:

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

WHAT YOU WILL BE DOING:

We are seeking a highly motivated Agentic/ML System Co-op who is pursuing a master’s or PhD degree in Computer Science, Electrical Engineering, or a related field and has a strong interest in machine learning systems and agentic AI. In this role, you will collaborate with our research and engineering teams to explore techniques that improve the efficiency, scalability, and performance of open-weight large language models on modern hardware architectures.

• Work with state-of-the-art open-weight large language models (LLMs) and leverage frontier coding agents to accelerate research and development.
• Contribute to agentic system design and software and hardware optimizations for machine learning workloads across AMD GPUs, vLLM, SGLang, and PyTorch.
• Collaborate with ML scientists, engineers, and architects in a dynamic, fast-paced environment that reflects the rapidly evolving AI ecosystem.
• Document your findings and contribute to technical reports, presentations, and potentially research publications.

WHO WE ARE LOOKING FOR:

• Currently enrolled in a master’s or PhD program in Computer Science, Electrical Engineering, or a related field at a U.S.-based university; PhD candidates are preferred.
• Proficiency in Python and/or C++, with exposure to performance profiling or optimization.
• A general understanding of machine learning software stacks and large language models, including frameworks such as PyTorch or JAX and coding-agent harnesses.
• A builder’s mindset, strong motivation, and intellectual curiosity, along with the ability to explore unfamiliar areas and incorporate feedback effectively.
• Demonstrated knowledge or experience in at least one of the following areas, preferably through relevant projects:
  • Agentic software engineering, agent-harness development and evolution, or agent and skill evaluation.
  • Performance analysis, GPU architecture, GPU kernel development and debugging, or machine learning compilers.

Benefits offered are described: AMD benefits at a glance.

This role is not eligible for visa sponsorship.

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.