Generative AI Engineer - Graphics

Advanced Micro Devices, Inc (AMD) is seeking an experienced AI/ML Engineer to build and optimize state-of-the-art Generative AI capabilities for image generation, diffusion-based learning, and multimodal applications. The role emphasizes scalable training and inference pipelines, rigorous model evaluation, and practical integration of AI advances into AMD technologies with GPU acceleration.

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 using 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 on emerging advances 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 building 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 constructing 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.

Required Education

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

Technologies

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

Location

On-site in the following locations: Austin, TX; Boston, MA; Orlando, FL; San Jose, California; Markham, ON, Canada.