AI Software Architect, Training and Optimization

Advanced Micro Devices, Inc

96d

About The Position

At AMD, our mission is to build great products that accelerate next-generation computing experiences – from AI and data centers, to PCs, gaming and embedded systems. Grounded in a culture of innovation and collaboration, we believe real progress comes from bold ideas, human ingenuity and a shared passion to create something extraordinary. When you join AMD, you’ll discover the real differentiator is our culture. We push the limits of innovation to solve the world’s most important challenges—striving for execution excellence, while being direct, humble, collaborative, and inclusive of diverse perspectives. Join us as we shape the future of AI and beyond. Together, we advance your career.

Requirements

• Minimum 4 years of experience required.
• Strong programming skills in C++ and Python.
• Strong development experience in at least one major DL framework such as Pytorch or Tensorflow in inference, fine tuning and/or training.
• Experience analyzing and optimizing the performance of AI software.
• Understanding of hardware bottlenecks and performance optimization.
• Experience in training in LLMs.

Nice To Haves

• Experience in multi-node training of LLM's.
• Expertise in profiling tools across the AI SW Stack (Torchprofiler, RocM profiler, Vtune, Nsight).
• Experience in implementing and optimizing parallel methods on GPU accelerators (NCCL/RCCL, OpenMP, MPI).
• Performance analysis skills for both CPU and GPU.
• Experience with Singularity, Docker, and/or Kubernetes.
• Experience in HPC.
• Strong bonus to have experience in MLPerf training workloads.

Responsibilities

• Enable software for world class datacenters and supercomputers.
• Optimize the software ecosystem for the next generation of GPU computational accelerators.
• Work with industry’s most sophisticated clients to leverage the latest hardware capabilities for their AI use cases.
• Combine the newest hardware with the industry’s latest applications, libraries, frameworks, and SDKs.

Benefits

• AMD benefits at a glance.

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level
• Education Level: Master's degree
• Number of Employees: 5,001-10,000 employees