Generative AI Engineer - Graphics

Advanced Micro Devices, Inc • Austin, TX

1d • Onsite

About The Position

AMD is seeking an experienced AI/ML Engineer to join its Applied Research and Engineering team, advancing Generative AI technologies. The role develops and optimizes AI models for image generation, diffusion-based learning, and multimodal applications in visual computing. The engineer works with researchers, software engineers, and hardware architects to build AI solutions using AMD GPU technologies for content creation, graphics, and intelligent computing.

The ideal candidate is passionate about Generative AI and machine learning, comfortable in a fast-paced research and engineering environment, and able to translate research into production-scale solutions through strong software development and cross-functional collaboration. This is an on-site role based in Austin, Boston, Orlando, San Jose, or Markham, Ontario. The role is not eligible for visa sponsorship.

Requirements

• Strong software development experience in Python and C++.
• Deep understanding of machine learning, deep learning, and neural network architectures.
• 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.
• Strong communication, collaboration, and problem-solving skills.

Nice To Haves

• 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 leveraging AI coding assistants, agentic AI systems, or automated development workflows.
• Master's degree or PhD in Computer Science, Artificial Intelligence, Machine Learning, Computer Engineering, or a related technical field preferred.

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 contribute to performance analysis and improvements.
• Stay current with advancements in Generative AI, image synthesis, diffusion models, and large-scale machine learning.

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level