Graphics and Machine Learning Inference Engineer

AMD is a technology company focused on building products that accelerate next-generation computing experiences, including AI, data centers, PCs, gaming, and embedded systems. The role involves working on ML performance for Gaming and Generative AI within AMD's product ecosystem.

• Strong programming skills in C++ and Python
• Hands-on experience optimizing GPU kernels for ML workloads using frameworks like HIP, CUDA, OpenCL, or HLSL
• Deep understanding of GPU architecture including compute units, memory hierarchy, cache, scheduling, and GPU programming model
• Experience with graphics APIs such as DirectX, Vulkan, OpenGL and shader programming
• Solid grasp of ML operators, their mathematical foundations, and modern ML architecture concepts
• Experience with concurrent programming and threading
• Development experience on Windows and/or Linux
• Familiarity with software tools such as CMake and GitHub
• Familiarity with low-level GPU languages (e.g., PTX)
• Understanding of ML techniques applied to graphics and rendering
• Strong communication, collaboration, and problem-solving skills
• Bachelor’s or Master’s degree in Computer Science, Computer Engineering, Electrical Engineering, or related field

• Transform neural network models and algorithms written in PyTorch/Onnx to efficient GFX shaders using languages such as HIP, CUDA, HLSL
• Design, implement, and optimize high-performance GPU kernels for ML operators
• Build AI-driven automation tools to accelerate kernel generation and optimization workflows
• Analyze and debug performance bottlenecks in collaboration with compiler and performance modeling teams
• Stay up-to-date with advancements in GPU architecture, ML accelerators, rendering techniques, and ML/GFX frameworks
• Contribute to tools and methodologies for integrating optimized shaders into game engines
• Drive next-generation features across hardware, drivers, and compilers focusing on performance per watt and performance per area
• Document and share knowledge on best practices for GFX and ML programming within the team
• Participate in code reviews and provide constructive feedback to peers

AMD and its subsidiaries are equal opportunity, inclusive employers and consider all applicants without regard to legally protected characteristics. AMD may use Artificial Intelligence to help screen, assess or select applicants for this position. AMD does not accept unsolicited resumes from headhunters, recruitment agencies, or fee-based recruitment services.