Spring/Summer 2027 PhD AI Agentic/ML System Co-op
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.
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
- System Thermal Design Staff Engineer5d agoPenang, Malaysia · Engineering
- PD Engineer5d agoHybrid (Bangalore) · Engineering
- PD Engineer5d agoBangalore · Engineering
- Low-Power & CDC Verification Engineer5d agoSanta Clara, California · Engineering
- Technical Program Manager5d agoShanghai · Engineering