Neural Rendering Research Inference Engineer – Advanced Graphics Programs jobs in United States
cer-icon
Apply on Employer Site
company-logo

AMD · 2 months ago

Neural Rendering Research Inference Engineer – Advanced Graphics Programs

AMD is a leading company in next-generation computing experiences, focusing on AI, data centers, and gaming. They are seeking a Neural Rendering Research Inference Engineer to design and optimize high-performance GPU kernels for machine learning applications, collaborating with various teams to enhance rendering speed and efficiency.

AI InfrastructureArtificial Intelligence (AI)Cloud ComputingComputerEmbedded SystemsGPUHardwareSemiconductor
check
Growth Opportunities
check
H1B Sponsor Likelynote

Responsibilities

Work 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
Design, implement and optimize high performance GPU kernels for ML operators
Work across research, hardware, driver and compiler teams to analyze and troubleshoot performance issues, provide solutions to improve rendering speed and ML workload efficiency
Stay current with latest advancements in GPU hardware, rendering techniques, graphics APIs and GPU accelerated ML
Contribute to the design and development of tools and methodologies for optimized shader integration to game engines
Document and share knowledge on best practices for GPU programming (both graphics and compute/ML) within the team
Participate in code reviews and provide constructive feedback to peers

Qualification

C++CUDAHIPGPU programmingNeural network operatorsGPU architectureMachine LearningEffective communicationProblem-solving skillsInterpersonal skills

Required

Deep technical expertise in translating neural network models and algorithms to efficient inference solutions
Strong programming expertise in C++, HIP, CUDA, HLSL and GPU-related technologies
Ability to effectively communicate and work optimally with different teams across AMD
Experience in designing, implementing and optimizing high performance GPU kernels for ML operators
Experience working with ML researchers and engineers to translate neural network models and algorithms written in PyTorch/Onnx to efficient GPU shaders
Ability to analyze and troubleshoot performance issues and provide solutions to improve rendering speed and ML workload efficiency
Knowledge of latest advancements in GPU hardware, rendering techniques, graphics APIs and GPU accelerated ML
Ability to document and share knowledge on best practices for GPU programming (both graphics and compute/ML)
Participation in code reviews and providing constructive feedback to peers
Undergrad degree required

Preferred

Strong object-oriented programming background, C/C++ preferred
Proven experience in developing and optimizing GPU kernels for machine learning workloads (eg:, 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
Understanding Machine Learning techniques and its application within graphics
Effective communication and problem-solving skills
Good interpersonal skills
Master's degree or PhD in Computer Science especially Graphics and/or Machine Learning, Computer Engineering, or equivalent preferred

Benefits

AMD benefits at a glance

Company

Advanced Micro Devices is a semiconductor company that designs and develops graphics units, processors, and media solutions.

H1B Sponsorship

AMD has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (836)
2024 (770)
2023 (551)
2022 (739)
2021 (519)
2020 (547)

Funding

Current Stage
Public Company
Total Funding
unknown
Key Investors
OpenAIDaniel Loeb
2025-10-06Post Ipo Equity
2023-03-02Post Ipo Equity
2021-06-29Post Ipo Equity

Leadership Team

leader-logo
Lisa Su
Chair & CEO
linkedin
leader-logo
Mark Papermaster
CTO and EVP
linkedin
Company data provided by crunchbase