AMD

4 months ago

AI Software Engineer(GPU Kernel)

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI

Last checked: 10 hours agoStatus: Live

Job Summary

Core member of the AMD AI software team optimizing and developing deep learning frameworks for AMD GPUs. You will enhance GPU kernels, design and optimize deep learning models, and improve training/inference performance across multi-GPU and multi-node systems, collaborating with internal GPU library teams and open-source maintainers, using cutting-edge compiler technologies in a Linux environment.

Location: Shanghai

Workplace: Onsite

Employment Type: Full time

Job Function: Software Engineering

Key Responsibilities

•Optimize Deep Learning Frameworks: Enhance and optimize frameworks like TensorFlow and PyTorch for AMD GPUs in open-source repositories.
•Develop GPU Kernels: Create and optimize GPU kernels to maximize performance for specific AI operations.
•Develop & Optimize Models: Design and optimize deep learning models specifically for AMD GPU performance.
•Collaborate with GPU Library Teams: Work closely with internal teams to analyze and improve training and inference performance on AMD GPUs.
•Collaborate with Open-Source Maintainers: Engage with framework maintainers to ensure code changes are aligned with requirements and integrated upstream.

Key Requirements

•Strong experience in C++ development within Linux environments
•GPU kernel development and optimization for AMD GPUs using HIP, CUDA, and assembly (ASM)
•Deep learning framework integration experience (TensorFlow/PyTorch)
•Knowledge of AMD architectures (GCN, RDNA) and low-level programming
•5+ years of professional software development experience with GPU optimization and framework development

Experience:5+ yearsAMD GPUsGPUDeep learning

Education:Master's

Skills:Problem-solvingCollaborationMentorshipCode reviewsCommunication

Languages:English

Tech Stack:C++LinuxHIPCUDAAssemblyCKCUTLASSTritonTensorFlowPyTorchLLVMROCmPython

Company Brief

AMD

Designs and produces semiconductor products including CPUs, GPUs, and adaptive SoCs for consumer, enterprise, and embedded markets, competing across PCs, data centers, and gaming industries.

Industry: Electronics Manufacturing

Company Size: Enterprise (1,001+ employees)

Revenue: USD 1B+

Growth: Public Company

Valuation: Public Company (Market Cap in USD)

Funding: IPO / Publicly Listed

Headquarters: Santa Clara, United States

Founded: 1969

Glassdoor: 3.9

Website LinkedIn

Loading job details...

Preparing the role view and application actions.

AMD