Spring/Summer 2027 Masters AI Training Software & Performance Co-op - CA, San Jose, United States
Originally posted 28 October 2026 by the employer.
Develop tools, profile workloads, and optimize distributed AI training on AMD Instinct™ GPU platforms.
About the role
AMD is seeking AI Software Engineers to join the team developing and optimizing software for AMD Instinct™ GPU accelerators. The role focuses on profiling, analyzing, and tuning large-scale AI distributed training and fine-tuning workloads across hardware and software stacks.
What you'll do
- Develop tools for debugging distributed training at a large scale. - Analyze and optimize the performance of AI software workloads across training and fine-tuning. - Identify hardware and software bottlenecks across CPUs, GPUs, memory systems, and communication infrastructure. - Profile and optimize performance across the AI software stack using advanced performance analysis tools. - Benchmark, evaluate, and tune AI frameworks, libraries, SDKs, and applications on AMD platforms.
What you'll need
- Current enrollment in a master's program in Computer Science, Computer Engineering, Artificial Intelligence, Machine Learning, or a related field. - Strong programming skills in C++ and/or Python with Agentic coding experience. - Development experience with at least one major deep learning framework such as PyTorch, JAX, TensorFlow, VLLM, or SGLang. - Expertise in profiling tools across the AI software stack including PyTorch Profiler, ROCm Profiler, VTune, and Nsight. - Experience implementing and optimizing parallel methods on GPU accelerators (NCCL/RCCL, OpenMP, MPI). - Deep understanding of LLM models, architectures, and various parallelisms (DP/TP/EP/CP) and their corresponding collectives.
Nice to have
- Publications in peer-reviewed AI, Machine Learning, High Performance Computing, or Computer Architecture conferences and journals.
Skills: ROCm, GPU, PyTorch, Performance Analysis, C++
This role has been open 0 days — well below the 56-day median for Infrastructure/Platform Software roles.
Infrastructure/Platform Software · GPU Software Stack
|
Open roles in category
247
|
Median days open
56 d
|
Median salary
$221k
|
See the full market breakdown ▾Category comparison, skills in demand, and who else is hiring
| Metric | AMD | All employers we track in this specialty (247 roles · 20 employers) |
|---|---|---|
| Open roles in this specialty | 51 | 247 |
| Open roles in the wider Software, Firmware & Systems family | 385 | 6389 · 147 employers |
| Median days open | 57 d | 56 d (+1 d vs this employer) |
| Median salary (USD postings) | — | $221k |
Skills observed across this category: ROCm, GPU, PyTorch, Performance Analysis, C++
Who's hiring in this category
- NVIDIA · 126 open roles · median 57 d
- AMD (this employer) · 49 open roles · median 56 d
- Intel Corporation · 27 open roles · median 15 d
- Qualcomm · 17 open roles · median 187 d
- Arm Holdings · 5 open roles · median 125 d
- Graphcore · 4 open roles · median 7 d
How we counted: 247 open Infrastructure/Platform Software (GPU Software Stack) roles from 20 employers tracked in the SemiconductorJobs index, counted 28 Oct 2026. Specialty figures count only roles carrying this exact specialty label, so an employer's related work in neighbouring specialties is not included there — it is counted in the wider Software, Firmware & Systems family row. Figures refresh nightly.