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

AMD
San Jose, Santa Clara
Workplace: HybridInternshipUSD 91,520 - 137,280 annuallyFunction: Software EngineeringEducation: phdSkills: []

Join AMD’s AI ecosystem as a PhD intern/co-op focused on AI model optimization and software engineering. You’ll develop, benchmark, and improve training, fine-tuning, and inference across CPU/GPU/accelerator platforms, profiling workloads and eliminating bottlenecks. Work on kernels and performance improvements using GPU programming models and contribute to AI frameworks, runtimes, distributed systems, and automation for evaluation and CI/CD.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
AMD
AMD
2 days ago

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

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 3 hours agoStatus: Live

Job Summary

Join AMD’s AI ecosystem as a PhD intern/co-op focused on AI model optimization and software engineering. You’ll develop, benchmark, and improve training, fine-tuning, and inference across CPU/GPU/accelerator platforms, profiling workloads and eliminating bottlenecks. Work on kernels and performance improvements using GPU programming models and contribute to AI frameworks, runtimes, distributed systems, and automation for evaluation and CI/CD.
Location: San Jose, Santa Clara
Workplace: Hybrid
Employment Type: Internship
Job Function: Software Engineering
Seniority: Intern level

Key Responsibilities

  • •Develop, benchmark, and optimize AI software for training, fine-tuning, and inference across CPU, GPU, and accelerator platforms.
  • •Profile AI workloads to 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 technologies such as HIP, CUDA, OpenCL, Triton, or related accelerator programming models.
  • •Research and implement AI model optimization methods including quantization, low-precision inference, sparsity, pruning, distillation, and parameter-efficient fine-tuning.
  • •Contribute to AI frameworks, libraries, execution runtimes, deployment technologies, and automated evaluation and containerized CI/CD workflows.

Pay and Benefits

Salary: USD 91,520 - 137,280 annually

Key Requirements

  • •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 in AI model inference, training, fine-tuning, or performance optimization.
  • •Knowledge or experience with GPU/accelerator kernel development and/or quantization, low-precision inference, sparsity, pruning, or distillation.
Experience:PhD
Education:PhD / Doctorate
Languages:English
Tech Stack:PythonC/C++HIPCUDAOpenCLTritonPyTorchTensorFlowJAXONNX RuntimeVLLMSGLangGitCMakeMakeCondaDockerCI/CD

Eligibility

Work Authorization:Authorization required. Sponsorship not provided.

Company Brief

AMD
Designs and produces semiconductor products including CPUs, GPUs, and adaptive SoCs for consumer, enterprise, and embedded markets, competing across PCs, data centers, and gaming industries.
Industry: Electronics Manufacturing
Company Size: Enterprise (1,001+ employees)
Revenue: USD 1B+
Growth: Public Company
Valuation: Public Company (Market Cap in USD)
Funding: IPO / Publicly Listed
Headquarters: Santa Clara, United States
Founded: 1969
Glassdoor
Glassdoor: 3.9
WebsiteLinkedIn