20 hours ago
San Jose, CA, USAEntry Level
Base Salary
$172k - $258k/yr
Responsibilities
- Implement, test, and debug features in collective communication libraries and networking software under senior-engineer guidance.
- Profile AI workloads on single-node and multi-node systems and identify performance bottlenecks.
- Run performance experiments across AMD GPU generations and network configurations and summarize results.
- Contribute to performance measurement, regression tracking, and reporting tools and infrastructure.
- Support model enablement and out-of-box performance across AI frameworks and help integrate changes with internal teams and open-source communities.
- Follow code review, testing, documentation, and other software engineering practices.
- Develop expertise in collectives, networking, and GPU performance through mentorship and hands-on project work.
Requirements
- Bachelor’s or master’s degree in computer science, computer engineering, electrical engineering, or equivalent.
- Professional software development experience with exposure to performance engineering, systems software, networking, or GPU programming.
- Proficiency in C++ and Python, with experience debugging, profiling, and writing tests.
- Understanding of collective communication concepts such as all-reduce and all-to-all, plus familiarity with RCCL, NCCL, or MPI.
- Foundational knowledge of GPU architecture; experience with HIP, CUDA, or similar GPU programming is preferred.
- Familiarity with PyTorch, JAX, vLLM, or SGLang and distributed training or inference concepts.
- Strong problem-solving, learning, collaboration, and written and verbal communication skills.
Benefits
- The role is onsite in San Jose, California or Austin, Texas.
- AMD benefits are offered, with details referenced in the AMD benefits at a glance materials.
About AMD
AMD designs and sells CPUs, GPUs, and adaptive/embedded computing products for PCs, data centers, gaming, and edge devices. Its portfolio includes Ryzen and EPYC processors, Radeon and Instinct graphics, and adaptive SoCs from its Xilinx acquisition, sold to OEMs, cloud providers, and device makers. Founded in 1969 and headquartered in Santa Clara, it is a public company on NASDAQ and supplies semi-custom chips for major game consoles.