AI Application Engineer

at  Advanced Micro Devices Inc

PREFERRED EXPERIENCE:

• Demonstrated experience with training and inference workloads on GPUs
• Executing applications in common frameworks (Pytorch, Tensorflow, Jax).
• Any experience with popular AI repositories (e.g. HuggingFace, MLperf) and understanding derived performance/functionality differences between them
• Proven understanding of baseline testing of popular AI applications (e.g. Resnet, BERT, DLRM, etc) and other real world codes such as Large Language Models, image detection, NLP, recommendation engines etc
• Understanding system level hardware design and its impact on performance
• Understand how the software stack affects performance: frameworks, precision, compilers, libraries, and other accompanying middleware applications.
• Customer-facing experience. Able to write technical documents and communicate at an appropriate level depending on the audience.
• Some Linux administration; understanding setup for HPC/AI middleware.

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_
Responsibilities:

PLEASE NOTE- THIS ROLE IS NOT CONSIDERED AN ENTRY LEVEL ROLE, AND INTERNS AND/OR NEW GRADUATES WITH ADVANCED DEGREES WILL NOT BE CONSIDERED. THIS ROLE REQUIRES SIGNIFICANT PROFESSIONAL ENGINEERING WORKING EXPERIENCE COMBINED WITH AN ADVANCED TECHNICAL DEGREE.

We are seeking an experienced AI Application Engineer to join the HPC Centre of Excellence (HPC CoE). The role broadly involves;
Working with customers and partners in support of RFP-driven requests. Providing hands-on support to customers and partners to enable their applications to run on AMD hardware and provide expected performance.
Broaden engineering investigations to understand both performance and characteristic performance across popular and customer-specific training and inference workloads. Understand competitive positioning.
Creating a body of technical documentation of AI performance on AMD hardware to support the Field Application team, partners, and customers.
This is a ‘hand-on’ technical role and we are looking for an individual with an established background in AI who is already familiar in executing and tuning training and inference workloads. In addition to the technical aspect, you will also need to be able to create and deliver presentations and training both remotely and in person to our customers and partners; therefore this will require some global travel.
AMD continues to ramp its in-house AI expertise, and as such this role provides an opportunity for the candidate to grow within the role while creating a significant impact on the wider business, with visibility to a senior level.
The role is initially advertised as located in North America, but it is also open to suitable candidates already based within Europe.

KEY RESPONSIBILITIES:

• Support winning new AI business. Enabling customers to execute their AI workloads on AMD GPUs and CPUs (principally ‘Instinct’ and ‘EPYC’ respectively). Supporting partners in RFP responses by testing requested workloads.
• Engineering: execute popular and customer-driven AI inference and training workloads, generate results and create a characteristic understanding of AI performance on AMD hardware. Understand how system and software choices affect performance. Compare performance to our competition.
• Run training and inference performance investigations using common frameworks (Pytorch, Tensorflow, JAX) and repositories (MLperf, Hugging Face etc.)
• Build a body of documentation for internal and external dissemination: AMD-internal guides, whitepapers, tuning guides, training collateral.
• Liaise and advise customers and partners through Proof of Concepts, presentations, and training.
• Assisting or lead effort to port applications, e.g. to different frameworks, change elements within the software
• Creating scripts and tools to enable a ‘fast start’ for customers and developers.
• Active engagement across AMD teams: GPU Business Unit, Engineering, Architecture, Platform, Software, and Product Development teams providing feedback and leadership from the field on requirements.
• Assist in creating TCO models to assist pricing with bid desk.
• Technically owning and resolving customer and partner issues. Submitting JIRA tickets and driving resolution.
• Automating repeatable procedures.