Neural 3D Graphics Software Engineer (SMTS)

Location: Santa Clara, California, US

Company: Advanced Micro Devices

Apply now

Apply for Job

What you do at AMD changes everything 

At AMD, we push the boundaries of what is possible.  We believe in changing the world for the better by driving innovation in high-performance computing, graphics, and visualization technologies – building blocks for gaming, immersive platforms, and the data center. 

Developing great technology takes more than talent: it takes amazing people who understand collaboration, respect, and who will go the “extra mile” to achieve unthinkable results.  It takes people who have the passion and desire to disrupt the status quo, push boundaries, deliver innovation, and change the world.   If you have this type of passion, we invite you to take a look at the opportunities available to come join our team.

The Role: 

We are looking for exceptional software engineers with a passion and demonstrated ability to rapidly learn new methods (from open-source code, academic papers, etc.), develop prototypes, invent new algorithms and help put the best ones in production. 


To be successful in this role, you need to be highly motivated and passionate about programming. You must love solving technical problems, think outside the box, and be versatile enough to go where the challenges take you. 


The Person: 

  • Proven track-record programming complex software in languages such as Python 3.x, Julia, C++ or CUDA 

  • Very strong aptitude for problem solving 

  • Works effectively both independently and in a team environment 

  • Configuring and troubleshooting your own Linux or Windows PC 

  • Effective written and verbal communication skills 

  • Ability to organize and manage priorities 


Key Responsibilities: 

This team contributes to multiple projects that combine real-time 3D graphics and machine learning.  

Depending on your level of experience and preference, you will share responsibilities such as: 


  • R&D for a new high-performance Neural 3D Rendering project, prototyped in Julia, with many opportunities to contribute to open-source repositories. 

  • Prototype and integrate state-of-the-art algorithms, from papers to integration into game engines, e.g. using Pytorch, JAX or Julia. Depending on the project, this may also include profiling the workload, doing in-depth optimization of the training and/or inference, interfacing with MLIR/ONNX, etc. 

  • R&D and ML Operations (MLOps). That project uses Pytorch, python, fastai, nbdev, Azure, etc. 

  • Collaborate and share insights to improve our GPUs with other teams, e.g. GPU hardware and software architects 

  • All projects offer many opportunities to learn, implement, integrate, optimize or invent new state-of-the ML and 3D algorithms 


Preferred Experience: 

  • ML fundamentals and state-of-the-art algorithms, theory and practice, e.g.: CNNs and other common neural network architectures, super-resolution, transformers, diffusion models,  PyTorch, JAX, ROCM/CUDA, computer vision, efficient architectures, etc.  

  • 3D graphics algorithms, APIs and game development, e.g.: DirectX, Vulkan, HLSL, SPIR-V, ray tracing, Unreal Engine, etc. In particular, experience creating 3D tools or integrating new 3D  graphics features in the Unreal Engine 4/5 (using C++) would be highly appreciated. 

  • Neural Rendering algorithms, e.g. Neural Radiance Fields (NeRF), Neural Light Fields, differentiable rasterization, etc. 

  • Portfolio of original, technically-challenging personal programming projects, or significant contribution to open-source projects 

  • MLOps or DevOps experience, e.g. cloud (Azure, AWS), Docker, Linux networking, system-wide  profiling/debugging 

  • Experience with complementary technologies and practices, e.g. LLVM/MLIR, in-depth  performance profiling, GPU architecture, 3D Content creation, etc. 



  • University degree in Computer Engineering, Computer Science or equivalent 



  • Proximity to an AMD development office, e.g.: Markham, Canada; Santa Clara, Boston, Orlando, San Diego, Austin, etc.; Munich, Germany (flexible) 

  • The team works in Europe or North-American time zones



Requisition Number: 182587 
Country: United States State: California City: Santa Clara 
Job Function: Design

Benefits offered are described here.

AMD does not accept unsolicited resumes from headhunters, recruitment agencies or fee based recruitment services. AMD and its subsidiaries are equal opportunity employers. We consider candidates regardless of age, ancestry, color, marital status, medical condition, mental or physical disability, national origin, race, religion, political and/or third party affiliation, sex, pregnancy, sexual orientation, gender identity, military or veteran status. Please click here for more information.

Apply now

Apply for Job