nvshmem4py-cu13

What it is and what it does

NVSHMEM4Py is a Python wrapper around NVIDIA's NVSHMEM library, which provides symmetric heap memory semantics for GPU-to-GPU communication. It allows Python developers to leverage NVSHMEM's collective operations and inter-GPU synchronization primitives without writing CUDA C/C++ code directly. The package depends on the CUDA runtime, nvidia-nvshmem-cu13 system library, cuda-python, numpy, and Cython, making it tightly coupled to NVIDIA's GPU ecosystem.

The package is actively maintained and follows NVIDIA's NVSHMEM SLA. It targets modern Python versions (3.9+) and provides prebuilt wheels for common Linux architectures. Installation requires both the CUDA 13 runtime and the underlying NVSHMEM library to be present on the system, which adds setup complexity but avoids runtime compilation for most users.

Use it for:

• Implementing multi-GPU collective algorithms (all-reduce, broadcast) without writing CUDA kernels.
• Building distributed GPU applications that need low-latency inter-GPU synchronization on a single node.
• Prototyping GPU communication patterns in Python before optimizing critical paths in CUDA.
• Integrating NVSHMEM operations into existing Python-based GPU frameworks or data-parallel workflows.

Worth the install?

AI-flagged interpretation of the facts on this page — verify before relying

Provides a Python interface to NVSHMEM, enabling GPU-accelerated symmetric heap memory operations for inter-GPU communication on NVIDIA systems.

Yes, if you are developing multi-GPU applications on NVIDIA hardware and need inter-GPU communication primitives. The package is actively maintained, has no known vulnerabilities, and offers a Pythonic entry point to NVSHMEM. However, installation requires CUDA 13 and the nvidia-nvshmem-cu13 library to be pre-installed, and the unclear license status warrants verification before use in proprietary code.

Install

nvshmem4py-cu13 on PyPI

pip install nvshmem4py-cu13

uv add nvshmem4py-cu13

poetry add nvshmem4py-cu13

Installing nvshmem4py-cu13

Before you install

Medium install friction: requires CUDA 13 runtime, nvidia-nvshmem-cu13 system library, and Cython compilation. Prebuilt wheels available for Python 3.10–3.13 on x86_64 and aarch64 Linux. Active maintenance with recent release.

License in practice

License status is unclear—no SPDX identifier or raw license text is available in the package metadata. Verify licensing terms before use in proprietary or redistributed code.

Quickstart

pip install nvshmem4py-cu13
import nvshmem4py_cu13
# Use NVSHMEM operations via the Python interface

Requires NVIDIA CUDA 13 runtime, nvidia-nvshmem-cu13 library, and compatible GPU hardware; Linux x86_64 or aarch64 only.

Verify before relying

• Specific NVSHMEM API coverage and which operations are exposed through the Python interface.
• Whether this package is suitable for single-GPU or only multi-GPU workloads.
• Performance characteristics and typical use-case scale (e.g., node-local vs. multi-node).
• Compatibility with non-NVIDIA GPU frameworks or heterogeneous compute environments.

Package facts

Evidence: nvshmem4py_cu13-0.3.1-cp310-cp310-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_34_aarch64.whl; nvshmem4py_cu13-0.3.1-cp310-cp310-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_34_x86_64.whl; nvshmem4py_cu13-0.3.1-cp311-cp311-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_34_aarch64.whl; nvshmem4py_cu13-0.3.1-cp311-cp311-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_34_x86_64.whl; nvshmem4py_cu13-0.3.1-cp312-cp312-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_34_aarch64.whl; nvshmem4py_cu13-0.3.1-cp312-cp312-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_34_x86_64.whl; nvshmem4py_cu13-0.3.1-cp313-cp313-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_34_aarch64.whl; nvshmem4py_cu13-0.3.1-cp313-cp313-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_34_x86_64.whl

Tags

More Distributed Computing packages

Further reading