This repository implements multiple generative diffusion frameworks (EDM, Consistency Models, etc.). It also implements some architectures (U-Net, Diffusion Transformers, etc.).

Derf (Dynamic erf) - Normalization-Free Transformer Activation. Reimplementation of arXiv:2512.10938

Port of TOF-MRA to CTA image-to-image translation in C

TQ-DiT: Efficient Time-Aware Quantization for Diffusion Transformers

Torchsmith is a minimalist library that focuses on understanding generative AI by building it using primitive PyTorch operations

DiT-VTON: Exploring Diffusion Transformer Framework for Multi-Category Virtual Try-On with Integrated Image Customization

Leverage SANA's capabilities using LitServe.

(ICCV 2025) 🎨 Lay-Your-Scene: Natural Scene Layout Generation with Diffusion Transformers

Flux Schnell diffusion transformer model fine tuning across hardware configurations

DiffInk: Glyph- and Style-Aware Latent Diffusion Transformer for Text to Online Handwriting Generation

A diffusion transformer implementation in Flax

InvarDiff: Cross-Scale Invariance Caching for Accelerated Diffusion Models

Minimal DDPM/DiT-based generation of MNIST digits

Unofficial LeetArxiv Implementation of the paper Scalable Diffusion Models with Transformers

Pytorch and JAX Implementation of Scalable Diffusion Models with Transformers | Diffusion Transformers in Pytorch and JAX

[NeurIPS2024 (Spotlight)] "Unified Gradient-Based Machine Unlearning with Remain Geometry Enhancement" by Zhehao Huang, Xinwen Cheng, JingHao Zheng, Haoran Wang, Zhengbao He, Tao Li, Xiaolin Huang

Whisfusion: Parallel ASR Decoding via a Diffusion Transformer

This repo implements Video generation model using Latent Diffusion Transformers(Latte) in PyTorch and provides training and inference code on Moving mnist dataset and UCF101 dataset

Implementation of Latent Diffusion Transformer Model in Tensorflow / Keras