DiT architecture