更新于 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)
- NVIDIA Dev Blog · 2026/4/22 23:50:10https://developer.nvidia.com/blog/simplify-sparse-deep-learning-with-universal-sparse-tensor-in-nvmath-python/