Senior GPU Kernel Engineer for AI Frameworks

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

Benefits offered by this job

Medical coverage

Inclusive workplace

Mentorship opportunities

Job summary

Advanced Micro Devices in Santa Clara is seeking an Industry Leading Expert C++ Developer to enhance deep learning frameworks for AMD GPUs. You will develop and optimize GPU kernels while collaborating with internal and open-source teams, utilizing cutting-edge compiler technologies to drive improvements.

The ideal candidate should have advanced C++ and Python proficiency, with a Master’s or PhD in a related field. Benefits include competitive medical coverage and a supportive work environment.

Qualifications

• Industry-leading expert in C++ development with deep knowledge of Linux environments.
• Extensive experience optimizing GPU kernels for AI operations.
• Proficient in deep learning model design and development for AMD GPUs.

Responsibilities

• Optimize and enhance deep learning frameworks like TensorFlow and PyTorch.
• Develop and optimize GPU kernels for AI tasks.
• Collaboration with GPU library teams for performance analysis.

Skills

C++ Development

Deep Learning Frameworks

GPU Kernel Optimization

Python

High-Performance Computing

Software Engineering Best Practices

Collaborative Work

Education

Master’s or PhD in Computer Science

Tools

TensorFlow

PyTorch

CUDA

LLVM

Job description

Advanced Micro Devices in Santa Clara is seeking an Industry Leading Expert C++ Developer to enhance deep learning frameworks for AMD GPUs. You will develop and optimize GPU kernels while collaborating with internal and open-source teams, utilizing cutting-edge compiler technologies to drive improvements.

The ideal candidate should have advanced C++ and Python proficiency, with a Master’s or PhD in a related field. Benefits include competitive medical coverage and a supportive work environment.