AI Systems & GPU Performance Engineer Intern

Advanced Micro Devices in San Jose, CA is seeking a highly motivated PhD AI Systems & GPU Performance Engineering Intern. You will optimize AI models, training workflows, and inference on AMD Instinct GPUs using ROCm, PyTorch, JAX, and related tools.

The role blends profiling, benchmarking, and end-to-end optimization to advance next-generation AI tech. Join a team that explores cutting-edge AI systems research, collaborates across disciplines, and shapes performance improvements on AMD's latest