← 返回信息流
更新于 2026/04/22 23:50

nvmath-python v0.9.0 集成通用稀疏张量,简化稀疏深度学习开发

NVIDIA 在 nvmath-python v0.9.0 中集成通用稀疏张量(UST),允许开发者将张量的稀疏模式与内存布局解耦,提升灵活性与性能。

UST 支持多种稀疏格式(COO、CSR、CSC 等)的统一接口,并自动选择最优 kernel 执行稀疏运算,无需手动调优。该设计面向科学计算与深度学习中的稀疏算子加速场景。

对于依赖稀疏计算(如 GNN、推荐系统、科学模拟)的工程团队,UST 降低了稀疏张量编程的复杂度,有望成为 GPU 上稀疏运算的标准抽象层。

速读

NVIDIA 在 nvmath-python v0.9.0 中集成通用稀疏张量 UST

相关源 (1)