NumPy & SciPy for GPU

Safe rust wrapper around CUDA toolkit

An open collection of implementation tips, tricks and resources for training large language models

An open collection of methodologies to help with successful training of large language models.

Distributed and decentralized training framework for PyTorch over graph

Best practices & guides on how to write distributed pytorch training code

Federated Learning Utilities and Tools for Experimentation

Efficient Distributed GPU Programming for Exascale, an SC/ISC Tutorial

NCCL Fast Socket is a transport layer plugin to improve NCCL collective communication performance on Google Cloud.

A Julia wrapper for the NVIDIA Collective Communications Library.

Sample examples of how to call collective operation functions on multi-GPU environments. A simple example of using broadcast, reduce, allGather, reduceScatter and sendRecv operations.

Python Distributed Non Negative Matrix Factorization with custom clustering

High performance NCCL plugin for Bagua.

N-Ways to Multi-GPU Programming

NCCL Examples from Official NVIDIA NCCL Developer Guide.

use ncclSend ncclRecv realize ncclSendrecv ncclGather ncclScatter ncclAlltoall

Experiments with low level communication patterns that are useful for distributed training.

Blink+: Increase GPU group bandwidth by utilizing across tenant NVLink.

Summary of call graphs and data structures of NVIDIA Collective Communication Library (NCCL)

Installation script to install Nvidia driver and CUDA automatically in Ubuntu