GPU Kernel Maestro for AI Training & Inference

Advanced Micro Devices, Inc.; Bellevue (WA); On-site; USD 180,000 - 240,000; Full time

Benefits offered by this job

AMD benefits

Job summary

Advanced Micro Devices, Inc. in Bellevue, WA seeks a PMTS Software Development Engineer to develop and optimize low-level GPU kernels that accelerate inference and training of large machine learning models.

You will design distributed training across multiple GPUs, profile performance, and collaborate with ML researchers and engineers to deliver production-ready solutions. Requirements include a Master’s degree or higher in CS/Engineering and 5 years of related experience; alternates with a

Qualifications

• Master's degree or foreign equivalent in Computer Science, Computer Engineering, Electrical Engineering, Software Engineering, or related field.
• Five (5) years of experience in the job offered or closely related engineering role.
• Alternate combination: Bachelor's degree or foreign equivalent plus seven (7) years of progressive post-baccalaureate experience.

Responsibilities

• Develop and optimize low-level GPU kernels to accelerate inference and training of large ML models.
• Maximize computational efficiency and reduce execution time while preserving model accuracy.
• Design and implement strategies for distributed model training and inference across multiple GPUs and nodes.
• Address data and model parallelism; profile and optimize performance; identify bottlenecks.
• Apply model quantization to reduce memory and compute overhead for edge and cloud deployment.
• Develop benchmarks and testing procedures and ensure production-ready solutions.
• Collaborate with ML researchers, software engineers, and infrastructure teams; create documentation.

Skills

Verilog RTL

UVM testbench

VLSI Design

Perl

Python

TCL/TK

FPGA

Hardware design

Education

Master's degree or foreign equivalent in Computer Science, Computer Engineering, Electrical Engineering, Software Engineering, or related field.

Bachelor's degree or foreign equivalent in Computer Science, Computer Engineering, Electrical Engineering, Software Engineering, or related field + 7 years of progressive post-baccalaureate experience.

Job description

Advanced Micro Devices, Inc. in Bellevue, WA seeks a PMTS Software Development Engineer to develop and optimize low-level GPU kernels that accelerate inference and training of large machine learning models.

You will design distributed training across multiple GPUs, profile performance, and collaborate with ML researchers and engineers to deliver production-ready solutions. Requirements include a Master’s degree or higher in CS/Engineering and 5 years of related experience; alternates with a