An implementation of model parallel autoregressive transformers on GPUs, based on the DeepSpeed library.
-
Updated
May 6, 2024 - Python
An implementation of model parallel autoregressive transformers on GPUs, based on the DeepSpeed library.
Add a description, image, and links to the deepspeed-library topic page so that developers can more easily learn about it.
To associate your repository with the deepspeed-library topic, visit your repo's landing page and select "manage topics."