GPU Kernel & AI Frameworks Engineer (Open Source)

Advanced Micro Devices; Markham; On-site; CAD 80,000 - 120,000; Full time

Job summary

A leading technology company in Markham is seeking a skilled engineer to optimize and develop deep learning frameworks for AMD GPUs. The ideal candidate will have strong expertise in C++ within Linux environments, with a focus on optimizing GPU kernels, collaborating with internal teams, and engaging with the open-source community. This role offers a chance to contribute to cutting-edge AI acceleration technologies and involves a bachelor's or master's degree in a relevant field.

Benefits

Comprehensive benefits package

Qualifications

• Experience in designing and optimizing GPU kernels on AMD GPUs.
• Skilled in Python development and software engineering best practices.
• Knowledgeable in multi-GPU and multi-node systems.

Responsibilities

• Enhance and optimize deep learning frameworks for AMD GPUs.
• Create and optimize GPU kernels for specific AI operations.
• Collaborate with internal teams to improve GPU performance.
• Engage with open-source maintainers for code integration.

Skills

C++ development in Linux environments

Deep learning frameworks optimization

Proactive problem-solving

Written and verbal communication

Education

Bachelor's or Master's Degree in Computer Science or related field

Tools

HIP

CUDA

TensorFlow

PyTorch

Job description

A leading technology company in Markham is seeking a skilled engineer to optimize and develop deep learning frameworks for AMD GPUs. The ideal candidate will have strong expertise in C++ within Linux environments, with a focus on optimizing GPU kernels, collaborating with internal teams, and engaging with the open-source community. This role offers a chance to contribute to cutting-edge AI acceleration technologies and involves a bachelor's or master's degree in a relevant field.