Senior Framework Engineer - Diffusion Inference

WHAT YOU DO AT AMD CHANGES EVERYTHING

At AMD, our mission is to build great products that accelerate next-generation computing experiences-from AI and data centers, to PCs, gaming and embedded systems. Grounded in a culture of innovation and collaboration, we believe real progress comes from bold ideas, human ingenuity and a shared passion to create something extraordinary. When you join AMD, you'll discover the real differentiator is our culture. We push the limits of innovation to solve the world's most important challenges-striving for execution excellence, while being direct, humble, collaborative, and inclusive of diverse perspectives. Join us as we shape the future of AI and beyond. Together, we advance your career.

Senior Framework Engineer - Diffusion Inference

As a Framework Engineer for Diffusion Model Inference, you will design, build, and evolve a production-grade inference framework for Diffusion Transformers (DiTs) powering state-of-the-art image and video generation. You will focus on framework-level engineering-model integration, scalable parallel inference, kernel plumbing, packaging, testing, and release management-ensuring diffusion workloads run out-of-the-box with exceptional performance on modern GPU systems.

This role sits at the intersection of model architecture, distributed systems, and GPU software stacks (ROCm, runtimes, libraries). You will collaborate closely with kernel, compiler, and hardware teams, and support top-tier customers by converting real-world requirements into upstream features, optimized pipelines, and dependable releases.

Key ResponsibilitiesDevelop and maintain a diffusion inference framework for image/video generation with clean APIs and strong compatibility with widely used diffusion ecosystems (e.g., HuggingFace diffusers-style pipelines and model wrappers).

Own scalable parallel inference features for DiT workloads-single-node and multi-node-including:Sequence parallel approaches (e.g., Unified Sequence Para