PyTorch/XLA based SPMD implementation of the QwenImage image-gen pipeline to run on TPU v6e. - View it on GitHub