SR GPU Network Software Engineer - Communications Libraries

WHAT YOU DO AT AMD CHANGES EVERYTHING

We care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences – the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our mission is the AMD culture. We push the limits of innovation to solve the world’s most important challenges. We strive for execution excellence while being direct, humble, collaborative, and inclusive of diverse perspectives. 

AMD together we advance_

THE ROLE:

As a GPU Network Software Engineer with our NTSG - Network Technology Solutions Group, you will design, implement, and test features in communication libraries, middleware, and frameworks to provide best in class support for GPU applications running high performance computing and machine learning workloads at scale. You will work with technical experts within AMD, our partners, and the open-source community to implement these features as part of AMD’s open source ROCm stack for GPU computation.

THE PERSON:

You are accustomed to working in a dynamic, geographically distributed agile team, where partnership and collaboration are paramount. You possess excellent written and verbal communication skills, and strong attention to detail.  You are results-oriented and accustomed to tight deadlines and changing priorities.  Most importantly, you are constantly thinking of ways to improve performance of multi-node GPU applications.

KEY RESPONSIBILITIES:

• Design, implement, and test features to enhance GPU support in communication libraries, middleware and frameworks
• Benchmark, profile and optimize code to maximize performance of multi-node GPU applications
• Deliver high-quality code and documentation following best practices for open-source software development
• Work with key technical experts at our customers, across AMD, and with our industry partners in the Ultra Ethernet Consortium and Ultra Accelerator Link Consortium to advance scale out and scale-up software and hardware solutions.

PREFERRED EXPERIENCE:

• Strong background developing system software in C/C
• 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)

• Familiarity with GPU programming in HIP or CUDA
• In-depth knowledge of best practices in software development, including testing, profiling, debugging, documentation, version control, issue tracking, and planning
• Proven track record contributing to open-source projects

ACADEMIC CREDENTIALS:

• B.Sc. or B.Eng. degree in Computer Science, Electrical Engineering, or equivalent

Advanced degrees, such as M.Sc., M.Eng., Ph.D. are preferred.

LOCATION:

Santa Clara, CA

#LI-BW1

#LI-hybrid

Benefits offered are described:  AMD benefits at a glance.

AMD does not accept unsolicited resumes from headhunters, recruitment agencies, or fee-based recruitment services. AMD and its subsidiaries are equal opportunity, inclusive employers and will consider all applicants without regard to 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, or any other characteristic protected by law.   We encourage applications from all qualified candidates and will accommodate applicants’ needs under the respective laws throughout all stages of the recruitment and selection process.