Senior GPU Software Engineer — AI & Kernel Performance

AMD; United States; On-site; USD 150,000 - 210,000; Full time

Job summary

AMD is seeking a Senior Software Development Engineer to develop and optimize software for AI and GPU computing platforms. You will work across GPU kernel development, performance optimization, AI frameworks, runtime systems, and compiler technologies to deliver industry-leading performance.

The ideal candidate has strong experience in GPU programming, performance analysis, and systems software development, with familiarity in MLIR/LLVM-based compiler flows.

Qualifications

• Strong C/C++ development skills.
• Experience with HIP, CUDA, OpenCL, or other GPU programming models.
• Strong understanding of GPU architecture, parallel programming, memory hierarchy, synchronization, and performance optimization.
• Hands-on experience profiling and optimizing AI, machine learning, HPC, or graphics workloads.
• Experience with performance analysis tools and benchmarking methodologies.
• Knowledge of LLVM, MLIR, compiler optimization, or code generation techniques.
• Experience with AI frameworks such as PyTorch, ONNX Runtime, TensorFlow, or similar ecosystems.
• Strong problem-solving and debugging skills across software and hardware stacks.

Responsibilities

• Develop and optimize GPU kernels for AI and compute workloads.
• Analyze and improve GPU performance, including compute utilization, memory bandwidth, cache efficiency, and kernel execution.
• Collaborate with AI framework, runtime, library, and hardware teams to improve end-to-end workload performance.
• Profile, benchmark, and debug large-scale AI workloads on AMD GPU platforms.
• Support compiler-driven optimizations and code generation using LLVM and MLIR-based technologies.
• Improve performance analysis, testing, benchmarking, and automation infrastructure.
• Enable and optimize new GPU hardware capabilities through software innovation.
• Apply AI-assisted development tools to improve productivity, debugging, and optimization workflows.

Skills

C/C++

GPU programming

Performance analysis

Systems software

Profiling tools

Compiler technology

PyTorch/ONNX

MLIR/LLVM

Education

Bachelor's or Master's in CS/CE

Tools

HIP

CUDA

OpenCL

LLVM/MLIR

Job description

AMD is seeking a Senior Software Development Engineer to develop and optimize software for AI and GPU computing platforms. You will work across GPU kernel development, performance optimization, AI frameworks, runtime systems, and compiler technologies to deliver industry-leading performance.

The ideal candidate has strong experience in GPU programming, performance analysis, and systems software development, with familiarity in MLIR/LLVM-based compiler flows.