pytorch / gloo
Collective communications library with various primitives for multi-machine training.
☆1,294Updated this week
Alternatives and similar repositories for gloo:
Users that are interested in gloo are comparing it to the libraries listed below
- ☆582Updated 7 years ago
- Optimized primitives for collective multi-GPU communication☆3,690Updated this week
- common in-memory tensor structure☆983Updated 3 weeks ago
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,313Updated this week
- ATen: A TENsor library for C++11☆694Updated 5 years ago
- A domain specific language to express machine learning workloads.☆1,759Updated 2 years ago
- A tensor-aware point-to-point communication primitive for machine learning☆257Updated 2 years ago
- ☆372Updated 7 years ago
- NCCL Tests☆1,085Updated 2 weeks ago
- TVM integration into PyTorch☆452Updated 5 years ago
- The Tensor Algebra SuperOptimizer for Deep Learning☆710Updated 2 years ago
- A performant and modular runtime for TensorFlow☆761Updated 2 weeks ago
- Reference implementations of MLPerf™ training benchmarks☆1,664Updated 3 weeks ago
- ☆389Updated 2 years ago
- TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.☆937Updated this week
- A fast GPU memory copy library based on NVIDIA GPUDirect RDMA technology☆1,080Updated last month
- A common bricks library for building scalable and portable distributed machine learning.☆868Updated last week
- A simple memory manager for CUDA designed to help Deep Learning frameworks manage memory☆297Updated 6 years ago
- Benchmarking Deep Learning operations on different hardware☆1,083Updated 4 years ago
- A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.☆982Updated 7 months ago
- A GPipe implementation in PyTorch☆837Updated 9 months ago
- A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.☆804Updated this week
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,040Updated last year
- nGraph has moved to OpenVINO☆1,349Updated 4 years ago
- ☆1,658Updated 6 years ago
- Acceleration package for neural networks on multi-core CPUs☆1,687Updated 10 months ago
- Quantized Neural Network PACKage - mobile-optimized implementation of quantized neural network operators☆1,540Updated 5 years ago
- Efficient GPU kernels for block-sparse matrix multiplication and convolution☆1,039Updated last year
- High performance Cross-platform Inference-engine, you could run Anakin on x86-cpu,arm, nv-gpu, amd-gpu,bitmain and cambricon devices.☆533Updated 2 years ago
- The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs☆1,293Updated 3 weeks ago