mid Software Engineer ic Bachelor's · Posted Mar 2, 2026

About this role

AMD is hiring a mid-level Software Engineer based in Shanghai, China. The posting calls out experience with TensorFlow, PyTorch, Deep Learning, Linux. Listed education preference: a bachelor's degree or equivalent.

Role
Software Engineer
Function
software engineering
Level
mid
Track
Individual contributor
Location
Shanghai, China
Education
Bachelor's degree
Department
Engineering
Posted
Mar 2, 2026
AI Summary
Optimize deep learning frameworks like TensorFlow and PyTorch for AMD GPUs by developing GPU kernels, enhancing model performance, and collaborating with internal teams and open-source maintainers. Requires strong C++ expertise in Linux environments, deep learning knowledge, and ability to work independently and mentor junior engineers.

More roles at AMD

Sr Performance Analysis Lead - Workload & CPU Arch
Bangalore, India · senior
Embedded Systems
Emulation Engineer
Shanghai, China · mid
Python C Ruby
Lead Rendering Development Engineer.
Warsaw, Poland · senior
TensorFlow PyTorch Data Structures
Firmware Technical Leader (IC role)
Taipei, Taiwan · staff
Linux Frontend Development Embedded Systems
Lead Performance Engineer
Warsaw, Poland · senior
Python C# Linux
All AMD jobs →

Job description

from AMD careers


WHAT YOU DO AT AMD CHANGES EVERYTHING 

At AMD, our mission is to build great products that accelerate next-generation computing experiences—from AI and data centers, to PCs, gaming and embedded systems. Grounded in a culture of innovation and collaboration, we believe real progress comes from bold ideas, human ingenuity and a shared passion to create something extraordinary. When you join AMD, you’ll discover the real differentiator is our culture. We push the limits of innovation to solve the world’s most important challenges—striving for execution excellence, while being direct, humble, collaborative, and inclusive of diverse perspectives. Join us as we shape the future of AI and beyond.  Together, we advance your career.  




THE ROLE:  

As a core member of the team, you will play a pivotal role in optimizing and developing deep learning frameworks for AMD GPUs. Your strong experience will be critical in enhancing GPU kernels, deep learning models, and training/inference performance across multi-GPU and multi-node systems. You will engage with both internal GPU library teams and open-source maintainers to ensure seamless integration of optimizations, utilizing cutting-edge compiler technologies and advanced engineering principles to drive continuous improvement. 

 

 

THE PERSON: 

This is an excerpt. Read the full job description on AMD careers →
All software engineering jobs software engineering in Shanghai, China Jobs in Shanghai, China software engineering salaries software engineering career path
All AMD Jobs Browse software engineering roles mid positions