AI/ML Engineer – Generative AI

About the role

AMD is seeking an experienced AI/ML Engineer to join the Applied Research and Engineering team focused on advancing Generative AI technologies. The role involves developing and optimizing state-of-the-art AI models for image generation and multimodal applications. The ideal candidate will work alongside industry-leading researchers and engineers to create innovative AI solutions leveraging AMD's GPU technologies. This position offers the opportunity to contribute to the future of content creation and intelligent computing.

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 leveraging 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 emerging 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.
• Strong communication, collaboration, and problem-solving skills.

Benefits

AMD provides a competitive 'Total Rewards' package that focuses on financial growth, health, and work-life balance.

Location: San Jose, CA • Type: Full-time • Posted: 8/5/2026 • Compensation: USD $210,000/Yr. – $300,000/Yr. • Undergraduate with 2+ Years of Experience