[Hiring] Neural Rendering Research Inference Engineer @Advanced Micro Devices, Inc

Role Description

We are seeking an exceptional Neural Rendering Research Inference Engineer - Advanced Graphics Program who has deep technical expertise in translating neural network models and algorithms to efficient inference solutions. The ideal candidate should be passionate about the intersection of Graphics and machine learning with a profound understanding of common neural network operators, their mathematical foundations and computational requirements, coupled with strong programming expertise in C++, HIP, CUDA, HLSL and GPU-related technologies.

The role involves:

• Working with ML researchers and engineers to translate neural network models and algorithms written in PyTorch/Onnx to efficient GPU shaders using languages such as HIP, Cuda, HLSL.
• Designing, implementing and optimizing high performance GPU kernels for ML operators.
• Analyzing and troubleshooting performance issues across research, hardware, driver and compiler teams, providing solutions to improve rendering speed and ML workload efficiency.
• Staying current with the latest advancements in GPU hardware, rendering techniques, graphics APIs and GPU accelerated ML.
• Contributing to the design and development of tools and methodologies for optimized shader integration to game engines.
• Documenting and sharing knowledge on best practices for GPU programming (both graphics and compute/ML) within the team.
• Participating in code reviews and providing constructive feedback to peers.

Qualifications

• Undergrad degree required.
• Master's degree or PhD in Computer Science especially Graphics and/or Machine Learning, Computer Engineering, or equivalent preferred.

Requirements

• Strong object-oriented programming background, C/C++ preferred.
• Proven experience in developing and optimizing GPU kernels for machine learning workloads (e.g., using HIP, OpenCL, CUDA, HLSL).
• Ability to program in low level languages (x86 asm, SSE, ISA, Ptx, AMD assembly).
• Strong understanding of GPU architecture (compute cores, cache hierarchy, memory model), graphics APIs (DirectX, OpenGL, Vulkan, etc.), and shader programming.
• Solid understanding of common neural network operators, their mathematical foundations, and computational requirements.
• Experience with modern concurrent programming and threading APIs.
• Experience with Windows, Linux operating system development.
• Experience with software development processes and tools such as debuggers, source code control systems (GitHub) and profilers is a plus.
• Understanding Machine Learning techniques and its application within graphics.
• Effective communication and problem-solving skills.
• Good interpersonal skills.

Benefits

• AMD benefits at a glance.