Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch - View it on GitHub