AI Performance Architect: GPU Software for AI Apps

Advanced Micro Devices, Inc.; Santa Clara (CA); On-site; USD 150,000 - 230,000; Full time

Job summary

AMD is seeking a highly motivated AI Software Engineer to join our world-class team. You will help enable DL models, libraries, and applications for Instinct GPUs in both on-prem and cloud environments, working to push the boundaries of efficiency and performance.

You will collaborate with a team of engineers to optimize AI software, understand hardware bottlenecks, and leverage the latest tools and frameworks to deliver scalable solutions across data centers and emerging AI workloads.

Qualifications

• Strong programming skills in C++ and Python.
• Strong development experience in at least one major DL framework for inference, fine-tuning, or training.
• MS or PhD in Computer Science or Computer Engineering with years of related experience.
• Experience developing software and system-level performance optimizations with GPUs.

Responsibilities

• Enable DL models, libraries, and applications for Instinct GPUs in on-prem and cloud environments.
• Analyze and optimize the performance of AI software and identify bottlenecks.
• Collaborate with teammates to leverage the latest hardware and software stacks to push innovation.

Skills

C++

Python

DL framework experience

Performance optimization

Education

MS/PhD in CS/CE

Tools

Docker

Kubernetes

Singularity

Nsight

Job description

AMD is seeking a highly motivated AI Software Engineer to join our world-class team. You will help enable DL models, libraries, and applications for Instinct GPUs in both on-prem and cloud environments, working to push the boundaries of efficiency and performance.

You will collaborate with a team of engineers to optimize AI software, understand hardware bottlenecks, and leverage the latest tools and frameworks to deliver scalable solutions across data centers and emerging AI workloads.