Metadata-Version: 2.4
Name: nccl4py
Version: 0.4.1
Summary: NCCL4Py: Python bindings for NCCL
Author: NVIDIA Corporation
License-Expression: Apache-2.0
Project-URL: homepage, https://developer.nvidia.com/nccl
Project-URL: documentation, https://docs.nvidia.com/deeplearning/nccl/user-guide/docs/index.html
Project-URL: repository, https://github.com/NVIDIA/nccl
Project-URL: issues, https://github.com/NVIDIA/nccl/issues
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: Intended Audience :: End Users/Desktop
Classifier: Natural Language :: English
Classifier: Operating System :: POSIX :: Linux
Classifier: Topic :: Education
Classifier: Topic :: Scientific/Engineering
Classifier: Topic :: Software Development :: Libraries
Classifier: Programming Language :: Cython
Classifier: Programming Language :: Python :: 3 :: Only
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Classifier: Programming Language :: Python :: 3.14
Classifier: Programming Language :: Python :: Implementation :: CPython
Classifier: Environment :: GPU :: NVIDIA CUDA
Classifier: Environment :: GPU :: NVIDIA CUDA :: 12
Classifier: Environment :: GPU :: NVIDIA CUDA :: 13
Requires-Python: >=3.10
Description-Content-Type: text/x-rst
License-File: LICENSE.txt
Requires-Dist: packaging
Requires-Dist: numpy
Requires-Dist: cuda-core~=1.0
Requires-Dist: cuda-pathfinder<2.0.0,>=1.5.4
Provides-Extra: cu12
Requires-Dist: nvidia-nccl-cu12; extra == "cu12"
Requires-Dist: cuda-bindings~=12.0; extra == "cu12"
Requires-Dist: nvidia-cutlass-dsl~=4.6; extra == "cu12"
Provides-Extra: cu13
Requires-Dist: nvidia-nccl-cu13; extra == "cu13"
Requires-Dist: cuda-bindings~=13.0; extra == "cu13"
Requires-Dist: nvidia-cutlass-dsl[cu13]~=4.6; extra == "cu13"
Dynamic: license-file

*****************************************************
nccl4py: Pythonic NCCL Communication for GPU Clusters
*****************************************************

.. image:: https://img.shields.io/badge/NVIDIA-black?logo=nvidia
   :target: https://www.nvidia.com/
   :alt: NVIDIA

`nccl4py <https://github.com/NVIDIA/nccl>`_ bridges Python's simplicity with the performance of NVIDIA Collective Communications Library (NCCL), and provides a Pythonic interface to NCCL library's functionality. It enables Python applications to leverage NCCL's GPU-accelerated multi-GPU and multi-node communication capabilities for distributed computing workloads.

nccl4py follows the NCCL SLA. The details of the NCCL SLA is available `here <https://docs.nvidia.com/deeplearning/nccl/sla/index.html>`_.

* `Homepage <https://developer.nvidia.com/nccl>`_
* `Repository <https://github.com/NVIDIA/nccl>`_
* `Documentation <https://docs.nvidia.com/deeplearning/nccl/user-guide/docs/index.html>`_
* `Issue tracker <https://github.com/NVIDIA/nccl/issues>`_

``nccl4py`` is under active development. Feedback and suggestions are welcome!


Installation
============

For CUDA 12.x:

.. code-block:: bash

   pip install nccl4py[cu12]

For CUDA 13.x:

.. code-block:: bash

   pip install nccl4py[cu13]

