Generative AI Engineer - Graphics

Advanced Micro Devices, Inc is seeking an experienced AI/ML Engineer to help build and improve Generative AI capabilities for image generation and visual content. This role sits at the intersection of machine learning and high-performance GPU computing, with a focus on developing scalable training and inference pipelines and translating new research into AMD technologies.

Working in an on-site environment, you will contribute to diffusion-based learning and multimodal applications, while evaluating model quality, performance, and efficiency using modern benchmarks and techniques.

Responsibilities

• Develop, train, fine-tune, and optimize Generative AI models for image generation and visual content creation.
• Design and implement diffusion models, multimodal architectures, and advanced machine learning algorithms.
• Build scalable training and inference pipelines that leverage high-performance GPU computing.
• Evaluate model quality, performance, and efficiency using state-of-the-art techniques and benchmarks.
• Collaborate with research, software, and hardware teams to integrate AI innovations into AMD technologies.
• Optimize AI workloads for AMD GPU architectures and support performance analysis and improvements.
• Stay current with advancements in Generative AI, image synthesis, diffusion models, and large-scale machine learning.

Requirements

• Strong software development experience in Python and C++.
• Deep understanding of machine learning, deep learning, and neural network architectures.
• Experience developing and training diffusion models, image generation models, GANs, transformers, or multimodal AI systems.
• Hands-on experience with PyTorch, TensorFlow, JAX, or similar machine learning frameworks.
• Experience fine-tuning and deploying large-scale AI models.
• Familiarity with distributed training, model optimization, and GPU acceleration techniques.
• Experience building AI training and inference pipelines in production environments.
• Knowledge of computer vision, image synthesis, and multimodal learning techniques.
• Experience with MLOps, CI/CD workflows, Git, and modern software engineering practices.
• Experience leveraging AI coding assistants, agentic AI systems, or automated development workflows.
• Strong communication, collaboration, and problem-solving skills.

Education

• Master's degree or PhD in Computer Science, Artificial Intelligence, Machine Learning, Computer Engineering, or a related field.

On-Site Locations

• Austin, TX
• Boston, MA
• Orlando, FL
• San Jose, California
• Markham, ON, Canada

Technologies

• Python, C++, PyTorch, TensorFlow, JAX
• GANs, transformers
• Git, MLOps, CI/CD
• Distributed training, GPU acceleration, GPU computing