← All roles

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

Apply on company site
San Jose, CA$91.5kUNAVAILABLEPosted 2w ago

About the role

Co-op position at AMD focusing on agentic/ML system research and optimization for open-weight LLMs, with emphasis on software/hardware co-design on AMD GPUs and modern ML frameworks. Term: Spring/Summer 2027, 40 hours/week, onsite or remote.

Responsibilities

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

Qualifications

  • Currently enrolled in a 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.

Skills

Required:PythonC++Machine learning frameworks (PyTorch, JAX)
Nice to have:Agentic software engineering / agent harness development

Benefits

  • AMD benefits at a glance
Last updated Oct 22, 2026. Interested? Apply on the company site.
AMD logo

About AMD

together we advance_AI

AMD delivers leadership high-performance and adaptive computing solutions to advance data center AI, AI PCs, intelligent edge devices, gaming, & beyond.

AIData CenterHigh-Performance ComputingGamingEdge Devices

More open roles at AMD