Diffusion Transformer (DiT)