Simple DiT - V1