返回全部动态

NVIDIA 发布 nvmath-python v1.0,加速 Python 核心数学运算

原标题:Run High-Performance Core Math at Scale with NVIDIA nvmath-python

NVIDIA Technical Blog一手来源产品发布质量 81

AI 摘要

NVIDIA 发布了 nvmath-python v1.0 正式版,这是一个 Python 库,旨在让 Python 科学计算社区能够轻松访问 CUDA-X 数学库的高性能。该库支持 CPU、GPU 以及多 GPU 多节点分布式计算,并提供了通用和专用 API,以及独特的通用稀疏张量(UST)功能。它简化了从 CPU 到 GPU 的代码迁移,并支持与 NumPy、CuPy、PyTorch 等数组库的互操作。

以上摘要由 AI 生成,可能存在误差。事实请以原文为准。

正文节选

NVIDIA nvmath-python is a library designed to bridge the gap between the Python scientific community and NVIDIA CUDA-X math libraries. It gives Python users access to CUDA-X performance for common math operations without disrupting existing workflows. Depending on the API, operations can run on a CPU, CUDA-enabled GPU, or distributed multi-GPU, multi-node systems. With the general availability of nvmath-python v1.0, this post explores the library’s design and unique capabilities for accelerating


发布时间:2026-07-31 06:43
抓取时间:2026-08-02 00:27
来源机构:NVIDIA
阅读原文developer.nvidia.com