DeepSpeed offers sparse attention kernels, an instrumental technology to support long sequences of model inputs, whether for text, image, or sound. Compared with the classic dense Transformers, it powers an order-of-magnitude longer input sequence and obtains up to 6x faster execution with comparable accuracy. It also outperforms state-of-the-art sparse implementations with 1.5-3x faster execution. Furthermore, our sparse kernels support efficient execution of flexible sparse format and empower users to innovate on their custom sparse structures.
- Brief overview, see our press release.
- Detailed technology deep dive, see our blog post.
- Tutorial on how to use sparse attention, see our Sparse attention tutorial.
- The source code for our sparse attention kernels can be found in the DeepSpeed repo and BERT pre-training code using sparse attention can be found in the DeepSpeedExamples repo.