Home
Jobs

Sr. AI System Infrastructure Engineer

10 - 15 years

40 - 45 Lacs

Posted:19 hours ago| Platform: Naukri logo

Apply

Work Mode

Work from Office

Job Type

Full Time

Job Description

MTS SOFTWARE DEVELOPMENT ENGINEER

THE ROLE:

AMD is looking for a specialized software engineer who is passionate about improving the performance of key applications and benchmarks

.

You will be a member of a core team of incredibly talented industry specialists and will work with the very latest hardware and software technology.

THE PERSON:

The ideal candidate should be passionate about software engineering and possess leadership skills to drive sophisticated issues to resolution. Able to communicate effectively and work optimally with different teams across AMD.

KEY RESPONSIBILITIES:

  • Design, develop, and optimize algorithms for collective communication operations (e.g., All-Reduce, All-to-All, Broadcast) within AMDs RCCL.
  • Analyze and tune the performance of collective communication libraries on large-scale GPU clusters, focusing on latency, bandwidth, and scalability over high-speed network fabrics.
  • Integrate and validate RCCL with various network transport layers and protocols, such as UEC, RoCE (RDMA over Converged Ethernet), and custom interconnects.
  • Collaborate closely with hardware, driver, and machine learning framework teams to co-design and debug system-level performance issues.
  • Develop robust benchmarking and profiling tools to identify and resolve bottlenecks in the communication software stack.
  • Contribute to the upstream open-source RCCL project and stay current with the latest advancements in the field.
  • Provide expert guidance on GPU cluster network topology and configuration to maximize collective communication performance.

PREFERRED EXPERIENCE:

  • 10+ years of experience in software development with a strong focus on high-performance computing or distributed systems.
  • Highly proficient in C/C++ programming and debugging in a Linux environment.
  • Experience with performance analysis, profiling, and debugging of complex, distributed systems in a Linux environment.
  • Proven track record of optimizing software for specific hardware architectures.
  • Strong analytical and problem-solving skills, with a proven ability to diagnose and resolve complex performance issues.
  • Hands-on experience with AMD ROCm RCCL or similar GPU collective communication libraries (NVIDIA NCCL, MSICCL, oneCCL, Open MPI, MPICH, or other MPI implementations etc.), RoCEv2/RDMA is a huge plus.
  • Effective communication and problem-solving skills.

ACADEMIC CREDENTIALS:

  • Bachelor s or Masters degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent
#LI-PM2

Mock Interview

Practice Video Interview with JobPe AI

Start Machine Learning Interview
cta

Start Your Job Search Today

Browse through a variety of job opportunities tailored to your skills and preferences. Filter by location, experience, salary, and more to find your perfect fit.

Job Application AI Bot

Job Application AI Bot

Apply to 20+ Portals in one click

Download Now

Download the Mobile App

Instantly access job listings, apply easily, and track applications.

coding practice

Enhance Your Skills

Practice coding challenges to boost your skills

Start Practicing Now
Advanced Micro Devices, Inc
Advanced Micro Devices, Inc

Semiconductors

Sunnyvale

RecommendedJobs for You