← All roles

2027 PhD AI Model Optimization & Software Engineer Intern/Co-op

AMD· AI/Software Engineering
Apply on company site
San Jose, CA, Santa Clara, CA$91.5kUNAVAILABLEPosted 1mo ago

About the role

AMD internship/co-op program focused on AI model optimization and software engineering across AI workloads, frameworks, and accelerator platforms. Roles involve development, benchmarking, optimization, and collaboration on next-generation AI software capabilities.

Responsibilities

  • Develop, benchmark, and optimize AI software for training, fine-tuning, and inference across CPU, GPU, and accelerator platforms.
  • Profile AI workloads, identify hardware and software 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 programming 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 technologies 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 technologies that improve the efficiency and portability of AI workloads.
  • Build automated evaluation systems, containerized development 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 project stakeholders to advance next-generation AI software capabilities.

Qualifications

  • Currently enrolled in a U.S.-based PhD program in Computer Science, Computer Engineering, Artificial Intelligence, Machine Learning, Electrical Engineering, or a related technical field.
  • Programming experience in Python and/or C/C++.
  • Experience with one or more AI frameworks or runtimes, such as PyTorch, TensorFlow, JAX, ONNX Runtime, vLLM, or SGLang.
  • Experience, coursework, research, or project work in: AI model inference, training, fine-tuning, or performance optimization; quantization, low-precision computing, sparsity, pruning, or distillation; GPU or accelerator kernel development; AI frameworks, libraries, compilers, graph optimization, or execution runtimes; performance profiling, benchmarking, debugging, or workload analysis; parallel computing, distributed systems, or collective communications; MLOps, DevOps, automated evaluation, CI/CD, or containerized environments; AI-assisted coding, coding agents, or agentic development workflows.
  • Familiarity with software development tools such as Git, CMake, Make, Conda, Docker, or related development and build technologies.

Skills

Required:PythonC/C++
Nice to have:PyTorchTensorFlow

Benefits

  • AMD benefits at a glance (link provided in description)
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
Company profileWebsite

More open roles at AMD