Staff Software Development Eng

Advanced Micro Devices, Inc•Markham, ON

3d•Hybrid

About The Position

The AI revolution is based on ultra-high performance math libraries running on the world's most advance compute accelerators. The role will be to work at the frontier of computing, at the intersection of numeric programming and next-generation GPU architectures, to build the fastest GEMM and Tensor operations the world has ever seen. You will build code that runs in the fastest data centers and supercomputers, and which forms the foundation of every major AI model today.

Requirements

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.
You are constantly thinking of ways to improve performance of multi-node GPU applications.
B.Sc. or B.Eng. degree in Computer Science, Electrical Engineering, or equivalent

Nice To Haves

Strong background developing system software in C/C++
Experience with at least one of the following:
Strong experience with networking and storage
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
Advanced degrees, such as M.Sc., M.Eng., Ph.D. are preferred.

Responsibilities

Design, implement, and test features to enhance GPU support in communication libraries, middleware and frameworks
Deliver benchmarks, profile results 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 to advance scale out and scale-up software and hardware solutions.