Principal AI Cluster Performance Validation Engineer

Advanced Micro Devices, Inc•Austin, TX

1d•Onsite

About The Position

ADVANCE YOUR CAREER. ADVANCE THE WORLD. At AMD, we believe technology has the power to solve the world’s most important challenges. From advancing healthcare and scientific discovery to powering AI and the technologies people rely on every day, innovation at AMD is shaping the future. Whether you’re designing next-gen processors, enabling AI breakthroughs, or bringing leading edge products to market, every role at AMD contributes to something bigger — technology that moves the world forward. Join us and, together, we’ll advance your career. THE ROLE: We are seeking a highly motivated and skilled Principal AI Cluster Performance Validation Engineer to join our dynamic team. In this role, you will be at the forefront of optimizing and achieving peak performance for GPU clusters. The focus of this role is the RDMA networks used in AI Clusters, understanding data flows between GPU, NIC and cluster network. The ideal candidate will have a strong background in GPU architectures, parallel clustered computing, and hands-on experience in system level performance tuning and debug methodologies. THE PERSON: The team fosters and encourages continuous technical innovation to showcase successes as well as facilitate continuous career development. A seasoned professional who enjoys hands-on problem-solving. In this role, you’ll shape long-term strategy, drive feature enablement and jump in to tackle challenges head-on. You’ll have a direct impact on performance, automation, and validation, while staying ahead of industry trends to provide strategic insights to senior management. The person should be experienced in debugging complex HW/FW and clustered configurations.

Requirements

Proven experience in optimizing the performance of GPU clusters.
RDMA network configuration, troubleshooting and performance tuning.
Strong understanding of GPU architectures, parallel computing concepts, and network protocols.
Proficiency in scripting languages (e.g., Python, Bash) for automation and performance analysis.
Experience with system level performance analysis tools and methodologies for GPU clusters.
Analytical mindset with excellent problem-solving and debug skills.
Familiarity with cluster management tools and systems.
Excellent communication and collaboration skills for effective teamwork.
Linux kernel networking expertise
Machine learning and/or HPC system design

Responsibilities

Evaluate the scalability of GPU clusters by conducting thorough testing under various workloads, ensuring optimal performance across different cluster sizes, configurations, and networking technologies (RoCE)
Develop and execute comprehensive benchmarking strategies to assess baseline performance, analyze bottlenecks, and identify areas for improvement within GPU cluster environments.
Collaborate with hardware and software teams to enhance the overall performance of GPU clusters, focusing on aspects such as RDMA throughput, RoCE v2 network congestion, latency, and collective communications.
Utilize profiling tools and methodologies to analyze and identify performance bottlenecks, providing actionable insights for improvement.
Implement optimization strategies, including but not limited to protocol enhancements, load balancing techniques, and parallel processing optimizations.
Create detailed documentation of performance analysis, tuning efforts, and outcomes, providing clear and concise reports for internal teams and stakeholders.
Work closely with cross-functional teams, including hardware engineers, software developers, and system architects, to integrate performance improvements into the GPU cluster architecture.
Stay current with the latest developments in GPU architectures, parallel processing, and emerging technologies to drive continuous improvement in GPU cluster performance.