AI Models Engineer, Efficient Generative AI

at  AMD

Job Description

THE PERSON:

We are looking for a candidate who is deeply passionate about AI models and possesses strong engineering skills to tackle complex challenges. You should have experience in optimizing and accelerating NLP/Generative AI models. Effective communication and collaboration with diverse teams across AMD are essential for success in this role.

PREFERRED EXPERIENCE:

Project experiences on generative tasks. Familiar with deep learning framework, e.g., Pytorch/JAX/ONNX/TensorFlow.
Project experiences on model compression, quantization, and end-to-end inference optimization.
Strong knowledge of artificial intelligence, deep learning or machine learning.
Strong coding skills in Python required, C/C++ skills a plus.
Skilled in academic paper writing and technical innovation.
Additional plus if recent publications include conferences such as NeuRIPS, CVPR, ECCV/ICCV, ICML, ICLR, etc.
Experience with any of the following also a plus: LLMs, stable diffusion, NeRF, or text-to-video generation.

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:

Join our innovative team at AMD as an AI Models Software Engineer. We are seeking passionate individuals who are dedicated to optimizing, accelerating, and applying AI models. As a member of our dynamic team, you will have the opportunity to shape the future of AI model development and make a significant impact in various industries and applications.

KEY RESPONSIBILITIES:

Research, design, and implement novel methods for efficient Generative AI.
Algorithmic model optimization methods design including quantization, sparsity, NAS, etc.
Publish your work.
Collaborate with other team members and teams.