NVIDIA / nccl
Optimized primitives for collective multi-GPU communication
☆3,375Updated last week
Alternatives and similar repositories for nccl:
Users that are interested in nccl are comparing it to the libraries listed below
- NCCL Tests☆967Updated 2 weeks ago
- Collective communications library with various primitives for multi-machine training.☆1,253Updated 2 weeks ago
- oneAPI Deep Neural Network Library (oneDNN)☆3,677Updated this week
- [ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl☆1,690Updated last year
- A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs…☆2,086Updated this week
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,240Updated this week
- Open MPI main development repository☆2,229Updated this week
- CUDA Templates for Linear Algebra Subroutines☆5,999Updated last week
- A fast GPU memory copy library based on NVIDIA GPUDirect RDMA technology☆923Updated last month
- Reference implementations of MLPerf™ training benchmarks☆1,636Updated this week
- A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep lear…☆5,244Updated this week
- Reference implementations of MLPerf™ inference benchmarks☆1,274Updated this week
- CUDA Library Samples☆1,720Updated 3 weeks ago
- common in-memory tensor structure☆930Updated 3 months ago
- Source code examples from the Parallel Forall Blog☆1,257Updated 5 months ago
- CUDA Core Compute Libraries☆1,385Updated this week
- Samples for CUDA Developers which demonstrates features in CUDA Toolkit☆6,756Updated 5 months ago
- Open deep learning compiler stack for cpu, gpu and specialized accelerators☆11,930Updated last month
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,647Updated this week
- Low-precision matrix multiplication☆1,787Updated 11 months ago
- Transformer related optimization, including BERT, GPT☆5,981Updated 9 months ago
- AIStore: scalable storage for AI applications☆1,355Updated this week
- PyTorch extensions for high performance and large scale training.☆3,232Updated this week
- Automatically Discovering Fast Parallelization Strategies for Distributed Deep Neural Network Training☆1,749Updated this week
- A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.☆972Updated 3 months ago
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆2,847Updated this week
- C++ extensions in PyTorch☆1,044Updated 5 months ago
- A retargetable MLIR-based machine learning compiler and runtime toolkit.☆2,929Updated this week
- ☆1,790Updated last year
- ☆573Updated 6 years ago