ML and HPC Network Programming Researcher

AMD - Bellevue, WA

Full-time - Senior

Bellevue, WA

Computer and Electronic Product Manufacturing

About the position

At AMD, we are committed to transforming lives through our technology, and we are looking for a researcher to join our Research and Advanced Development team. This role focuses on inventing and implementing novel programming models and runtime environments specifically designed for high-performance computing (HPC) systems that utilize heterogeneous processors and accelerators. As a key member of our team, you will be responsible for developing innovative software solutions that demonstrate the capabilities of our cutting-edge hardware architectures. Your work will not only enhance the efficiency and performance of our systems but also contribute to defining the software stack for AMD's future hardware accelerators. In this position, you will engage in a variety of tasks, including the design and implementation of new features, APIs, and abstractions aimed at optimizing performance and improving programmer productivity. You will collaborate closely with teams across AMD Research and Advanced Development, product groups, and vendor partners to enhance the AMD HPC ecosystem. Your contributions will drive the results of research projects into product roadmaps, and you will have the opportunity to submit patentable inventions based on your findings. Clear communication of research results to academic, product, and commercial audiences will be essential, as you will need to articulate complex technical information effectively. We are looking for someone who is passionate about bringing new systems to life and is ready to tackle the challenges of high-performance computing. If you thrive in a collaborative environment and are eager to push the boundaries of innovation, we invite you to join us in this exciting journey.

Responsibilities

• Invent, design and implement novel programming models and runtime environments for high performance computing systems comprised of heterogeneous processors and accelerators.
• Implement innovative software solutions and demonstrate their effectiveness on prototype hardware.
• Work on software enhancements that aim to potentially improve programmer productivity.
• Drive the results of the research projects into product roadmaps.
• Collaborate with teams within AMD Research and Advanced Development, product groups, and vendor partners to help improve the AMD HPC ecosystem.
• Submit patentable inventions.
• Clearly communicate research findings to academic, product and commercial audiences.

Requirements

• PhD degree in Computer Engineering or Electrical Engineering preferred.
• Experience in developing and debugging GPU and/or multi-threaded CPU applications.
• Familiarity with ML/HPC/datacenter software, middleware, and device driver development.
• Strong skills in C/C++, Python, and/or GPU programming.
• Experience with networking simulation platforms (NS-3, OMNeT++, etc.).
• Experience with High Performance Computing networking.
• Familiarity with the AMD ROCm software stack.
• Version control systems such as Git.

Nice-to-haves

• Distributed system experience is desired.
• Prior exposure to ML frameworks, OpenSHMEM, or MPI programming models is preferred.
• Demonstrated experience in ideation, evaluation, and optimization of a research project.
• Excellent written and oral communication skills, ability to organize and present complex technical information.
• Strong analytical skills.

Benefits

• Employee stock purchase plan
• Annual bonus or sales incentive eligibility
• Competitive benefits package