AI/ML Engineer – Generative AI

Advanced Micro Devices, Inc•San Jose, CA

1d•Onsite

About The Position

AMD is seeking an experienced AI/ML Engineer to join our Applied Research and Engineering team focused on advancing the next generation of Generative AI technologies. In this role, you will develop and optimize state-of-the-art AI models for image generation, diffusion-based learning, and multimodal applications that push the boundaries of visual computing. You will work alongside industry-leading researchers, software engineers, and hardware architects to build innovative AI solutions that leverage AMD’s latest GPU technologies and accelerate the future of content creation, graphics, and intelligent computing.

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.
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 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.