https://mer.dev/posts/scalable-diffusion-models-with-transformers/