跳到正文
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 扩展 pytorch_block_sparse,其 BlockSparseLinear 可直接替换 torch.nn.Linear,让语言模型更小更快。该库基于 NVIDIA CUTLASS 的块稀疏矩阵乘法,75% 稀疏度下矩阵运算约比稠密快 2 倍、内存占用减少 4 倍,但当前整体性能仍比 cuBLAS 稠密实现慢约 2 倍。

来源:Hugging Face Blog · huggingface.co