Custom DiT diffusion policy (from-scratch training loop)

initial commit