Alternatives and similar repositories for pytorch-communication-benchmarks

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

• cchan / tccl
  extensible collectives library in triton
  ☆91Updated 8 months ago
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆122Updated last year
• deepspeedai / DeepSpeed-Kernels
  ☆71Updated 8 months ago
• meta-pytorch / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆309Updated 3 months ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆132Updated 6 months ago
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆84Updated last month
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆173Updated 8 months ago
• usyd-fsalab / fp6_llm
  An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).
  ☆272Updated 4 months ago
• microsoft / TileFusion
  TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.
  ☆104Updated 5 months ago
• meta-pytorch / kraken
  Triton-based Symmetric Memory operators and examples
  ☆65Updated last month
• gpu-mode / triton-index
  Cataloging released Triton kernels.
  ☆277Updated 3 months ago
• meta-pytorch / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆297Updated last week
• Deep-Learning-Profiling-Tools / triton-viz
  ☆257Updated this week
• Jokeren / triton-samples
  ☆28Updated 10 months ago
• stanford-futuredata / stk
  ☆113Updated last year
• CalvinXKY / mfu_calculation
  A simple calculation for LLM MFU.
  ☆50Updated 3 months ago
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆123Updated last year
• infinigence / FlashOverlap
  A lightweight design for computation-communication overlap.
  ☆194Updated 2 months ago
• meta-pytorch / BackendBench
  Ship correct and fast LLM kernels to PyTorch
  ☆125Updated 3 weeks ago
• ColfaxResearch / layout-categories
  This repository contains companion software for the Colfax Research paper "Categorical Foundations for CuTe Layouts".
  ☆82Updated 2 months ago
• AlibabaPAI / FLASHNN
  ☆102Updated last year
• triton-lang / kernels
  ☆96Updated last year
• microsoft / nnscaler
  nnScaler: Compiling DNN models for Parallel Training
  ☆120Updated 2 months ago
• alibaba / easydist
  Automated Parallelization System and Infrastructure for Multiple Ecosystems
  ☆81Updated last year
• meta-pytorch / KernelAgent
  Autonomous GPU Kernel Generation via Deep Agents
  ☆172Updated last week
• IBM / triton-dejavu
  Framework to reduce autotune overhead to zero for well known deployments.
  ☆90Updated 2 months ago
• NVIDIA / nvshmem
  NVIDIA NVSHMEM is a parallel programming interface for NVIDIA GPUs based on OpenSHMEM. NVSHMEM can significantly reduce multi-process com…
  ☆407Updated 3 weeks ago
• tspeterkim / paged-attention-minimal
  a minimal cache manager for PagedAttention, on top of llama3.
  ☆126Updated last year
• facebookexperimental / triton
  Github mirror of trition-lang/triton repo.
  ☆104Updated this week
• xlite-dev / HGEMM
  ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.
  ☆135Updated 7 months ago