Distributed training (multi-node) of a Transformer model - View it on GitHub