GitHub repository: kashif/torchscale
Description: Transformers at any scale