Minimal implementation of a diffusion transformer. Clean, readable code focused on the core DiT architecture without unnecessary abstractions.
Simple diffusion transformer
Minimal implementation of a diffusion transformer. Clean, readable code focused on the core DiT architecture without unnecessary abstractions.