NVIDIA 发布 nvmath-python v1.0,加速 Python 核心数学运算
原标题:Run High-Performance Core Math at Scale with NVIDIA nvmath-python
AI 摘要
NVIDIA 发布了 nvmath-python v1.0 正式版,这是一个 Python 库,旨在让 Python 科学计算社区能够轻松访问 CUDA-X 数学库的高性能。该库支持 CPU、GPU 以及多 GPU 多节点分布式计算,并提供了通用和专用 API,以及独特的通用稀疏张量(UST)功能。它简化了从 CPU 到 GPU 的代码迁移,并支持与 NumPy、CuPy、PyTorch 等数组库的互操作。
正文节选
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