AI Performance Software Engineer — GPU Optimization

Advanced Micro Devices; Santa Clara (CA); On-site; USD 100,000 - 140,000; Full time

Benefits offered by this job

AMD benefits at a glance

Job summary

Advanced Micro Devices is seeking a motivated AI Software Engineer to join their cutting-edge AI Software Solutions Team in Santa Clara, California. In this role, you will enable deep learning models for Instinct GPUs and help optimize software for advanced computational accelerators.

Candidates should have strong programming skills in C++ and Python, with experience in deep learning frameworks such as PyTorch or Tensorflow. A minimum of 4 years of relevant experience is required. The position is integral to supporting AMD's mission of building innovative AI solutions.

Qualifications

• Minimum 4 years of experience required.
• Experience analyzing and optimizing AI software performance.
• Ability to work in team and independently.

Responsibilities

• Enable deep learning models and libraries for Instinct GPUs.
• Optimize software ecosystem for GPU computational accelerators.
• Collaborate with clients to leverage hardware capabilities.

Skills

Strong programming skills in C++

Strong programming skills in Python

Experience with DL frameworks like PyTorch or Tensorflow

Excellent analytical and problem-solving skills

Education

MS or PhD in Computer Science or Computer Engineering

Tools

Profiling tools (Torchprofiler, RocM profiler)

Docker

Kubernetes

Job description

Advanced Micro Devices is seeking a motivated AI Software Engineer to join their cutting-edge AI Software Solutions Team in Santa Clara, California. In this role, you will enable deep learning models for Instinct GPUs and help optimize software for advanced computational accelerators.

Candidates should have strong programming skills in C++ and Python, with experience in deep learning frameworks such as PyTorch or Tensorflow. A minimum of 4 years of relevant experience is required. The position is integral to supporting AMD's mission of building innovative AI solutions.