AMD

1 day ago

GPU Kernel Performance Engineer

✓ Verified Job

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

Source: Company careers pageValidated by: Fursa AI

Last checked: 2 hours agoStatus: Live

Job Summary

Enhance and optimize machine learning (ML) kernels for better performance and efficiency on AMD’s ROCm Core team. You’ll collaborate with cross-functional partners to design and implement ML algorithms and frameworks, and contribute to high-quality software components. Responsibilities include developing scalable kernel performance, participating in code reviews, testing, and debugging, and staying current on ML and hardware optimization advances.

Location: Shanghai

Workplace: Onsite

Employment Type: Full time

Job Function: Healthcare (Clinical, Medical, Wellness)

Seniority: Entry level

Key Responsibilities

•Develop and optimize ML kernels focused on performance and scalability.
•Collaborate with cross-functional teams to understand requirements and deliver high-quality software components.
•Assist in design and implementation of new ML algorithms and frameworks.
•Participate in code reviews, testing, and debugging to deliver reliable software.
•Stay up to date with machine learning and hardware optimization advancements.

Key Requirements

•Bachelor’s degree or above in Computer Science, Electrical Engineering, or a related field.
•Strong foundation in machine learning concepts and algorithms.
•Experience with programming languages such as C++ and Python (or similar).
•Familiarity with GPU programming and parallel computing, including CUDA or ROCm.
•Experience with ML frameworks such as TensorFlow or PyTorch (or similar).

Education:Bachelor's in Computer Science, Electrical Engineering, or related field

Skills:Problem-solvingAttention to detailTeamworkCommunicationDiligence

Languages:En-us

Tech Stack:C++PythonCUDAROCmTensorFlowPyTorch

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