xdit-project/xDiT

Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"

A pipeline parallel training script for diffusion models.

A PyTorch-native inference engine with cache, parallelism, quantization and cpu offload for DiTs.

Scalable and memory-optimized training of diffusion models

Karras et al. (2022) diffusion models for PyTorch

[CVPR 2025 Oral] Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Models

https://wavespeed.ai/ Best inference performance optimization framework for HuggingFace Diffusers on NVIDIA GPUs.

SANA: Efficient High-Resolution Image Synthesis with Linear Diffusion Transformer

Implementation of Denoising Diffusion Probabilistic Model in Pytorch