Alternatives and similar repositories for pytorch-communication-benchmarks

Users that are interested in pytorch-communication-benchmarks are comparing it to the libraries listed below

• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆141Updated 8 months ago
• meta-pytorch / kraken
  Triton-based Symmetric Memory operators and examples
  ☆80Updated 2 weeks ago
• cchan / tccl
  extensible collectives library in triton
  ☆93Updated 10 months ago
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆125Updated last year
• infinigence / FUSCO
  High-performance distributed data shuffling (all-to-all) library for MoE training and inference
  ☆109Updated last month
• inclusionAI / asystem-amem
  A NCCL extension library, designed to efficiently offload GPU memory allocated by the NCCL communication library.
  ☆87Updated last month
• meta-pytorch / BackendBench
  Ship correct and fast LLM kernels to PyTorch
  ☆139Updated 2 weeks ago
• ademeure / DeeperGEMM
  DeeperGEMM: crazy optimized version
  ☆73Updated 8 months ago
• infinigence / FlashOverlap
  A lightweight design for computation-communication overlap.
  ☆213Updated last week
• deepspeedai / DeepSpeed-Kernels
  ☆71Updated 10 months ago
• meta-pytorch / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆314Updated 5 months ago
• alibaba / easydist
  Automated Parallelization System and Infrastructure for Multiple Ecosystems
  ☆82Updated last year
• meta-pytorch / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆319Updated this week
• Deep-Learning-Profiling-Tools / triton-viz
  ☆277Updated this week
• microsoft / TileFusion
  TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.
  ☆105Updated 7 months ago
• gpu-mode / triton-index
  Cataloging released Triton kernels.
  ☆291Updated 4 months ago
• AlibabaPAI / FLASHNN
  ☆105Updated last year
• open-lm-engine / accelerated-model-architectures
  A bunch of kernels that might make stuff slower 😉
  ☆75Updated last week
• flagos-ai / libtriton_jit
  A Triton JIT runtime and ffi provider in C++
  ☆30Updated this week
• HanGuo97 / hilt
  ☆39Updated last month
• foundation-model-stack / foundation-model-stack
  🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.
  ☆219Updated this week
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆127Updated last year
• ColfaxResearch / layout-categories
  This repository contains companion software for the Colfax Research paper "Categorical Foundations for CuTe Layouts".
  ☆95Updated 4 months ago
• IBM / triton-dejavu
  Framework to reduce autotune overhead to zero for well known deployments.
  ☆94Updated 4 months ago
• antgroup / DeepXTrace
  DeepXTrace is a lightweight tool for precisely diagnosing slow ranks in DeepEP-based environments.
  ☆91Updated 2 weeks ago
• flashinfer-ai / cutlass-viz
  ☆65Updated 9 months ago
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆88Updated this week
• abcdabcd987 / libfabric-efa-demo
  ☆76Updated last year
• Azure / msccl-executor-nccl
  ☆47Updated last year
• CalvinXKY / mfu_calculation
  A simple calculation for LLM MFU.
  ☆66Updated 4 months ago