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