跳到正文
Hugging Face Blog·· 2020-09-10AI 评分38

Hugging Face 开源 pytorch_block_sparse:让语言模型更小更快

Block Sparse Matrices for Smaller and Faster Language Models

AI 导读

Hugging Face 发布 pytorch_block_sparse 扩展库,基于 NVIDIA CUTLASS 实现块稀疏矩阵乘法,可作为 torch.nn.Linear 的直接替代,让神经网络更小更快。75% 稀疏度下速度约为稠密矩阵的 2 倍,内存占用降至 1/4;当前稀疏实现比 cuBLAS 优化的稠密版本慢约 2 倍,但已比 PyTorch 原生稀疏实现快一个数量级。

来源:Hugging Face Blog · huggingface.co