Diffusion Transformers (DiTs) trained on MNIST dataset - View it on GitHub