NVIDIA / nccl
Optimized primitives for collective multi-GPU communication
☆3,690Updated this week
Alternatives and similar repositories for nccl:
Users that are interested in nccl are comparing it to the libraries listed below
- NCCL Tests☆1,085Updated 2 weeks ago
- Collective communications library with various primitives for multi-machine training.☆1,294Updated this week
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,313Updated this week
- A fast GPU memory copy library based on NVIDIA GPUDirect RDMA technology☆1,080Updated last month
- oneAPI Deep Neural Network Library (oneDNN)☆3,783Updated this week
- A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper, Ada and Bla…☆2,390Updated this week
- [ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl☆1,745Updated last year
- Reference implementations of MLPerf™ training benchmarks☆1,664Updated 3 weeks ago
- CUDA Templates for Linear Algebra Subroutines☆7,391Updated this week
- CUDA Library Samples☆1,910Updated this week
- ☆582Updated 7 years ago
- A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep lear…☆5,378Updated this week
- C++ extensions in PyTorch☆1,083Updated 3 months ago
- Examples demonstrating available options to program multiple GPUs in a single node or a cluster☆697Updated 2 months ago
- CUDA Core Compute Libraries☆1,619Updated this week
- common in-memory tensor structure☆983Updated 3 weeks ago
- Compiler for Neural Network hardware accelerators☆3,289Updated 11 months ago
- nGraph has moved to OpenVINO☆1,349Updated 4 years ago
- Automatically Discovering Fast Parallelization Strategies for Distributed Deep Neural Network Training☆1,786Updated last week
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆3,127Updated this week
- A domain specific language to express machine learning workloads.☆1,759Updated 2 years ago
- Open MPI main development repository☆2,333Updated this week
- The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs☆1,293Updated 3 weeks ago
- Efficient GPU kernels for block-sparse matrix multiplication and convolution☆1,039Updated last year
- FlashInfer: Kernel Library for LLM Serving☆2,764Updated this week
- Source code examples from the Parallel Forall Blog☆1,283Updated 9 months ago
- Benchmarking Deep Learning operations on different hardware☆1,083Updated 4 years ago
- PyTorch extensions for high performance and large scale training.☆3,308Updated last week
- Reference implementations of MLPerf™ inference benchmarks☆1,369Updated last week
- A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.☆982Updated 7 months ago