DiT for VAE (and Video Generation) - View it on GitHub