AI Model Optimization Engineer Intern (Co-op)

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

Benefits offered by this job

Benefits at a glance

Job summary

Advanced Micro Devices, Inc. (AMD) is seeking highly motivated AI Model Optimization & Software Engineer Interns/Co-op for multiple opportunities in AI model optimization, framework engineering, performance analysis, GPU computing, distributed systems, and AI software infrastructure.

Locations include San Jose, CA or Santa Clara, CA; roles require full-time hours in hybrid or onsite structures for the co-op/intern term.

Qualifications

• Currently enrolled in a U.S.-based PhD program in Computer Science, Computer Engineering, AI, ML, EE, or related field.

Responsibilities

• Develop, benchmark, and optimize AI software for training, fine-tuning, and inference across CPU, GPU, and accelerator platforms.
• Profile AI workloads, identify bottlenecks, and implement performance improvements across compute, memory, communication, framework, and runtime layers.
• Design, prototype, and optimize GPU or CPU kernels using HIP, CUDA, OpenCL, Triton, or related accelerator models.
• Research and implement AI model optimization methods such as quantization, low-precision inference, sparsity, pruning, distillation, and parameter-efficient fine-tuning.
• Contribute to AI frameworks, libraries, execution runtimes, and deployment tech such as PyTorch, TensorFlow, JAX, ONNX Runtime, vLLM, or SGLang.
• Develop or evaluate parallel and distributed computing methods, including collective communication operations and scalable training or inference techniques.
• Explore compiler, graph optimization, kernel-generation, and runtime tech to improve AI workloads portability.
• Build automated evaluation systems, containerized environments, CI/CD pipelines, internal tools, and developer productivity solutions for AI workflows.
• Use AI-assisted development tools and coding agents to support software development, testing, debugging, and optimization.
• Collaborate with software engineers, researchers, architects, platform teams, and stakeholders to advance next-gen AI software capabilities.

Skills

Python

C/C++

PyTorch

TensorFlow

JAX

Git

Docker

CI/CD

Education

PhD candidate in CS/CE/AI

Tools

HIP

CUDA

OpenCL

Triton

vLLM

SGLang

Job description

Advanced Micro Devices, Inc. (AMD) is seeking highly motivated AI Model Optimization & Software Engineer Interns/Co-op for multiple opportunities in AI model optimization, framework engineering, performance analysis, GPU computing, distributed systems, and AI software infrastructure.

Locations include San Jose, CA or Santa Clara, CA; roles require full-time hours in hybrid or onsite structures for the co-op/intern term.