AI Performance Software Engineer – GPU & DL Optimizations

AMD; Santa Clara (CA); On-site; USD 120,000 - 160,000; Full time

Benefits offered by this job

Comprehensive benefits package

Inclusive culture

Career advancement opportunities

Job summary

AMD is seeking a highly motivated AI Software Engineer to join its team in Santa Clara, California. This role focuses on enabling and optimizing AI models for advanced GPU computational accelerators. Candidates should have strong programming skills in C++ and Python, along with experience in deep learning frameworks.

The ideal candidate will possess excellent problem-solving skills and a willingness to work collaboratively within a team. Minimum 4 years of experience required. Benefits include a comprehensive package, fostering an inclusive culture.

Qualifications

• Minimum 4 years of experience in software development.
• Strong understanding of AI software performance and optimization.
• Ability to work well within a team environment.

Responsibilities

• Enable DL models, libraries, and applications for Instinct GPUs.
• Analyze and optimize AI software performance.
• Work collaboratively with a team of Software Engineers.

Skills

Strong programming skills in C++ and Python

Development experience in DL frameworks like PyTorch or TensorFlow

Analytical and problem-solving skills

Experience in software and system-level performance optimizations

Education

MS or PhD in Computer Science or Computer Engineering

Tools

Profiling tools (Torchprofiler, RocM profiler, Vtune, Nsight)

Docker

Kubernetes

Job description

AMD is seeking a highly motivated AI Software Engineer to join its team in Santa Clara, California. This role focuses on enabling and optimizing AI models for advanced GPU computational accelerators. Candidates should have strong programming skills in C++ and Python, along with experience in deep learning frameworks.

The ideal candidate will possess excellent problem-solving skills and a willingness to work collaboratively within a team. Minimum 4 years of experience required. Benefits include a comprehensive package, fostering an inclusive culture.