2026 Masters Artificial Intelligence Frameworks Intern/Co-op

Advanced Micro Devices, Inc•San Jose, CA

5h•Hybrid

About The Position

At AMD, our mission is to build great products that accelerate next-generation computing experiences—from AI and data centers, to PCs, gaming and embedded systems. Grounded in a culture of innovation and collaboration, we believe real progress comes from bold ideas, human ingenuity and a shared passion to create something extraordinary. When you join AMD, you’ll discover the real differentiator is our culture. We push the limits of innovation to solve the world’s most important challenges—striving for execution excellence, while being direct, humble, collaborative, and inclusive of diverse perspectives. Join us as we shape the future of AI and beyond. Together, we advance your career. As an AMD intern, 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.

Requirements

You are currently enrolled in a US based University in a Master's degree program majoring in Computer Science, Computer Engineering, or a related field.
Strong programming ability in C++/CUDA and Python.
Experience with PyTorch, TensorFlow, vLLM, SGLang, or ONNX.
Familiarity with development tools such as git, cmake, make, conda, or docker.
Knowledge of data structures, algorithms, image processing, neural networks, and computer architecture.

Nice To Haves

Experience with AI framework development, operator implementation, ML quantization, or compilers is a plus.

Responsibilities

You will develop and benchmark AI software frameworks and tools.
Your responsibility will include operator development and model deployment across CPU/GPU/NPU stacks.
You will research inference quantization and acceleration for models such as LLMs, ViTs, and CNNs.
You will collaborate closely with development teams and project managers to drive performance improvements.