AI/ML Engineer – Generative AI

AMD is seeking an experienced AI/ML Engineer to join its Applied Research and Engineering team, focusing on advancing generative AI technologies such as image generation and diffusion models. The role involves developing and optimizing state-of-the-art models, building scalable training and inference pipelines on AMD GPUs, and collaborating with research, software, and hardware teams. This is an on-site position based in San Jose, California.

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

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