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

Advanced Micro Devices; Austin, Northern (TX, KY); Hybrid; USD 28,000 - 41,000; Full time

Job summary

AMD in Austin, Texas is offering AI Model Optimization & Software Engineer Internships/Co-ops. You will work across AI model optimization, framework engineering, performance analysis, GPU computing, distributed systems, and AI software infrastructure.

This role is suited for graduate students or PhD candidates participating in Spring/Summer or Summer/Fall terms, with 40 hours per week and a hybrid or onsite structure.

Qualifications

• Currently enrolled in a U.S.-based PhD program in Computer Science, Computer Engineering, AI, ML, EE, or related field.
• Proficiency in Python and/or C/C++.
• Experience with AI frameworks such as PyTorch, TensorFlow, JAX, ONNX Runtime, vLLM, or SGLang.

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, and runtime layers.
• Design, prototype, and optimize GPU or CPU kernels using HIP, CUDA, OpenCL, or Triton.
• Research and implement AI model optimization methods such as quantization, sparsity, pruning, and distillation.
• Contribute to AI frameworks, libraries, and execution runtimes such as PyTorch, TensorFlow, JAX, or ONNX Runtime.
• Develop or evaluate parallel and distributed computing methods for scalable training or inference.
• Explore compiler and graph optimization technologies to improve efficiency and portability.
• Build automated evaluation systems, CI/CD pipelines, and developer productivity tools for AI workflows.
• Use AI-assisted development tools to support coding, testing, debugging, and optimization.

Skills

Python

C/C++

AI frameworks

Education

PhD student (U.S.-based)

Tools

Git

CMake

Docker

Conda

Job description

ADVANCE YOUR CAREER. ADVANCE THE WORLD.

At AMD, we believetechnology has the power to solve the world’s most important challenges. From advancing healthcare and scientific discovery to powering AI and the technologies people rely on every day, innovation at AMDis shapingthefuture.

Whetheryou’redesigning next-gen processors, enabling AI breakthroughs, orbringing leading edge products to market, every role at AMD contributes to something bigger— technologythat moves the world forward.Join us and, together, we’ll advance your career.

JOB DETAILS:

• Location: Austin, Texas, USA
• Onsite/Hybrid: This role requires the student to work full time (40 hours a week), in either a hybrid or onsite work structure throughout the duration of the co-op/intern term
• Duration:
  • Spring/Summer Co-op: January 25, 2027 - August 13, 2027
  • Summer Internship:
    • Semester Students: May 24, 2027 - August 13, 2027
    • Quarter Students: June 21, 2027 - September 10, 2027
  • Summer/Fall Co-op:
    • Semester Students: May 24, 2027 - December 10, 2027
    • Quarter Students: June 21, 2027 - December 10, 2027

As an AMD intern and co-op, 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.

WHAT YOU WILL BE DOING:

We are seeking highly motivated AI Model Optimization & Software Engineer Interns/Co-op to join our teams. We are recruiting for multiple opportunities across AI model optimization, framework engineering, performance analysis, GPU computing, distributed systems, and AI software infrastructure.

Depending on your background, interests, and the needs of the team, you may:

• 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 technologies such as 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.

WHO WE ARE LOOKING FOR:

We recognize that candidates may bring strengths in different areas of AI Model Optimization & AI Software. You are not expected to have experience in every technology or technical area listed below. If you have experience, academic research, project work, or a strong technical interest in one or more of these areas, we encourage you to apply.

• 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 one or more of the following areas:
  • 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.
• Research publications, preprints, open-source contributions, or participation in the AI and machine learning developer community are preferred but not required.

If you are excited about advancing AI software and bring experience in at least one of the areas above, we encourage you to apply.

This role is not eligible for visa sponsorship.

AMD and its subsidiaries are equal opportunity, inclusive employers and will consider all applicants without regard to age, ancestry, color, marital status, medical condition, mental or physical disability, national origin, race, religion, political and/or third-party affiliation, sex, pregnancy, sexual orientation, gender identity, military or veteran status, or any other characteristic protected by law. We encourage applications from all qualified candidates and will accommodate applicants’ needs under the respective laws throughout all stages of the recruitment and selection process.

AMD may use Artificial Intelligence to help screen, assess or select applicants for this position. AMD’s “Responsible AI Policy” is available here.

This posting is for an existing vacancy.