Principal Software Development Eng. - GPU Communication Libraries, SHMEM/MPI jobs in United States
cer-icon
Apply on Employer Site
company-logo

AMD · 2 weeks ago

Principal Software Development Eng. - GPU Communication Libraries, SHMEM/MPI

AMD is a company dedicated to building innovative products that enhance next-generation computing experiences. The role involves designing software modules for GPU systems, improving existing codebases, and collaborating with architecture specialists to advance future products.

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

Responsibilities

Design software modules in C++, Python, HIP, assembly to enable collective communication software for datacenter GPU systems
Understand existing codebases and software designs, and make improvements to enhance maintainability and operational efficiency
Work with AMD’s architecture specialists to improve future products and plan software support strategies
Aid management in planning, and delivering industry-leading software
Stay informed of software and hardware trends and innovations, especially pertaining to software algorithms and hardware architecture
Design and develop new groundbreaking AMD technologies
Participate in new ASIC and hardware bring ups

Qualification

C++PythonGPU software developmentMPISHMEMAgile developmentSoftware performance evaluationLeadership skillsCommunication skills

Required

Design software modules in C++, Python, HIP, assembly to enable collective communication software for datacenter GPU systems
Understand existing codebases and software designs, and make improvements to enhance maintainability and operational efficiency
Work with AMD's architecture specialists to improve future products and plan software support strategies
Aid management in planning, and delivering industry-leading software
Stay informed of software and hardware trends and innovations, especially pertaining to software algorithms and hardware architecture
Design and develop new groundbreaking AMD technologies
Participate in new ASIC and hardware bring ups
Bachelor's or Master's degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent

Preferred

Experience with agile software development practices
Demonstrated capacity to technically lead developers of varying levels
Proficient in C/C++ & Python programming employing best software design practices
GPU software development involving HIP, CUDA, or OpenCL
Experience with at least one of the following: Implementing communication middleware like MPI/SHMEM, Implementing lower-level communication frameworks like UCX and libfabric, or development using RDMA APIs, Development and optimization of communication collective algorithms (e.g. AllReduce)
Experience in software performance evaluations, optimizations and debugging
Ability to closely interact with software technical leads, program managers, and interface with hardware teams

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