Senior GPU Kernel Optimization Engineer

AMD
Shanghai
Workplace: OnsiteFull timeCNY 411,670 - 588,100 annuallyFunction: Solutions Engineering & Sales EngineeringExperience: 4+ yearsEducation: bachelorsSkills: ["Performance optimization","Profiling","Collaboration","Programming"]

Design, develop, and optimize high-performance GPU kernels and AI operators for training and inference workloads. Use profiling and architecture-aware tuning (memory optimization, kernel fusion, parallelization) to improve kernel performance. Develop kernels with CUDA, HIP, and Triton, and build AI-powered coding agents for kernel generation, optimization, benchmarking, and validation. Collaborate with model and infrastructure teams to improve end-to-end AI system performance.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
AMD
AMD
17 hours ago

Senior GPU Kernel Optimization Engineer

✓ Verified Job

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

Source: Company careers pageValidated by: Fursa AI
Last checked: 13 hours agoStatus: Live

Job Summary

Design, develop, and optimize high-performance GPU kernels and AI operators for training and inference workloads. Use profiling and architecture-aware tuning (memory optimization, kernel fusion, parallelization) to improve kernel performance. Develop kernels with CUDA, HIP, and Triton, and build AI-powered coding agents for kernel generation, optimization, benchmarking, and validation. Collaborate with model and infrastructure teams to improve end-to-end AI system performance.
Location: Shanghai
Workplace: Onsite
Employment Type: Full time
Job Function: Solutions Engineering & Sales Engineering
Seniority: Mid level

Key Responsibilities

  • •Design, develop, and optimize high-performance AI operators and GPU kernels for training and inference workloads.
  • •Analyze and improve kernel performance using profiling, memory optimization, kernel fusion, parallelization, and architecture-aware tuning.
  • •Develop GPU kernels using CUDA, HIP, Triton, or similar programming frameworks.
  • •Build AI-powered coding agents and automation workflows for kernel generation, optimization, benchmarking, and validation.
  • •Collaborate with model and infrastructure teams to improve end-to-end AI system performance.

Pay and Benefits

Salary: CNY 411,670 - 588,100 annually

Key Requirements

  • •Bachelor's degree (or above) in Computer Science, Computer Engineering, Artificial Intelligence, or a related field, with 4+ years of relevant experience.
  • •Strong programming skills in C++ and/or Python.
  • •Hands-on experience developing and optimizing GPU kernels using CUDA, HIP, Triton, or similar frameworks.
  • •Solid understanding of GPU architecture and performance optimization (memory hierarchy, occupancy, register pressure, shared memory).
  • •Experience with performance profiling and optimization tools such as Nsight, rocprof, or equivalent.
Experience:4+ years
Education:Bachelor's in Computer Science, Computer Engineering, Artificial Intelligence, or related field
Skills:Performance optimizationProfilingCollaborationProgramming
Languages:En-us
Tech Stack:C++PythonCUDAHIPTritonNsightRocprofGEMMAttentionMoEEmbeddingLLMSFTRLHFDPOGRPOCUDA/HIP/Triton

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
Glassdoor: 3.9
WebsiteLinkedIn