NVIDIA / nccl
Optimized primitives for collective multi-GPU communication
☆3,564Updated 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,040Updated last week
- Collective communications library with various primitives for multi-machine training.☆1,277Updated this week
- CUDA Templates for Linear Algebra Subroutines☆7,150Updated this week
- A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs…☆2,293Updated this week
- oneAPI Deep Neural Network Library (oneDNN)☆3,747Updated this week
- A fast GPU memory copy library based on NVIDIA GPUDirect RDMA technology☆1,007Updated 2 weeks ago
- FlashInfer: Kernel Library for LLM Serving☆2,439Updated this week
- Reference implementations of MLPerf™ training benchmarks☆1,653Updated this week
- Transformer related optimization, including BERT, GPT☆6,089Updated 11 months ago
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,277Updated this week
- Open deep learning compiler stack for cpu, gpu and specialized accelerators☆12,134Updated this week
- common in-memory tensor structure☆963Updated last week
- TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.☆915Updated this week
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆3,037Updated this week
- Automatically Discovering Fast Parallelization Strategies for Distributed Deep Neural Network Training☆1,773Updated this week
- A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep lear…☆5,323Updated this week
- Open MPI main development repository☆2,292Updated this week
- The Triton Inference Server provides an optimized cloud and edge inferencing solution.☆8,926Updated this week
- [ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl☆1,735Updated last year
- Source code examples from the Parallel Forall Blog☆1,269Updated 7 months ago
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆11,350Updated last week
- PyTorch extensions for high performance and large scale training.☆3,278Updated 2 months ago
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,711Updated this week
- Low-precision matrix multiplication☆1,794Updated last year
- A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.☆979Updated 6 months ago
- ☆580Updated 6 years ago
- A domain specific language to express machine learning workloads.☆1,756Updated last year
- Tile primitives for speedy kernels☆2,170Updated this week
- CUDA Python: Performance meets Productivity☆1,212Updated this week
- CUDA Core Compute Libraries☆1,539Updated this week