SMTS Software Development Eng- ML Framework And Runtime

WHAT YOU DO AT AMD CHANGES EVERYTHING

We care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences - the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our mission is the AMD culture. We push the limits of innovation to solve the world’s most important challenges. We strive for execution excellence while being direct, humble, collaborative, and inclusive of diverse perspectives.

AMD together we advance_

THE ROLE

AMD is looking for a talented, self-driven and motivated engineer to technically lead AIG’s Vitis AI Compiler projects working on AMD’s XDNA (AI Engine) architecture and the Vitis AI family of software tools. The XDNA is an industry leading NPU (Neural Processing Engine) architecture in terms of performance per watt and is used in AMD’s client and embedded devices as the primary engine for Machine Learning workloads. It is the hardware engine behind Windows Co-pilot on AMD devices.

The team provides a fast-paced environment offering each of its members immense opportunity to interact with a wide variety of people including from other organizations like hardware designers, marketing, support, and even direct customer interaction, and truly learn and grow their skills and capabilities.

The Person

The ideal candidate should be passionate about software engineering and possess leadership skills to drive sophisticated technical issues to resolution. They should have demonstrated ability to identify technical problems, explore and propose viable options, and apply technical solutions. They should be able to excel in a global team environment with strong verbal and written communication skills.

Key Responsibilities

• Vitis AI is AMD’s primary SDK that enables users to compile and run their ML models on the XDNA architecture which forms the basis for AMD’s. As a senior member of this high-performance team, the selected candidate will have the opportunity to work on integrating the ML tool chain into frameworks like ONNX, Pytorch, TensorFlow etc.
• Candidate will have opportunity to work on orchestrating the compilation of ML model through different phases
• Candidate will integrate runtime execution of ML model on the NPU hardware through the runtime and driver.
• Candidate will collaborate with compiler and runtime teams to bring up latest AI models like CNNs, Transformers, Stable Diffusion, NLPs etc. on the XDNA simulator.
• Candidates would develop a deeper understanding of the various ML models, and how they are executed, identify performance bottlenecks and enable faster development.
  
  
  

• Minimum 10 years of relevant work experience.
• Strong background in large scale C++ based development and debug, including Design Patterns
• Experience with multi-threaded programming infrastructure and performance optimization
• Experience in the software development environment on both Linux and Windows is required.
• Experience in any one of the ML Framework like ONNX, Pytorch etc is strongly desired.
• Experience with scalable builds and code versioning through github, docker, CMake, artifactory is highly desired.
  
  
  

• Bachelor’s or Masters degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent