Alternatives and similar repositories for pytorch-direct

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

• jiazhihao / attention_superoptimizer
  An Attention Superoptimizer
  ☆21Updated 4 months ago
• zhuzilin / pytorch-malloc
  An external memory allocator example for PyTorch.
  ☆14Updated 3 years ago
• Lin-Mao / DrGPUM
  A memory profiler for NVIDIA GPUs to explore memory inefficiencies in GPU-accelerated applications.
  ☆25Updated 7 months ago
• sjtu-epcc / Tacker
  Tacker: Tensor-CUDA Core Kernel Fusion for Improving the GPU Utilization while Ensuring QoS
  ☆25Updated 3 months ago
• bytedance / QSync
  Official resporitory for "IPDPS' 24 QSync: Quantization-Minimized Synchronous Distributed Training Across Hybrid Devices".
  ☆20Updated last year
• facebookresearch / fairring
  Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …
  ☆65Updated 3 years ago
• microsoft / TileFusion
  TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.
  ☆88Updated last week
• LeiWang1999 / Stream-k.tvm
  ☆19Updated 8 months ago
• chips-compilers-mlsys-21 / chips-compilers-mlsys-21.github.io
  ☆11Updated 4 years ago
• lixiuhong / batched_gemm
  ☆38Updated 5 years ago
• zhisbug / Cavs
  Cavs: An Efficient Runtime System for Dynamic Neural Networks
  ☆14Updated 4 years ago
• Oneflow-Inc / dfccl
  ☆25Updated 3 months ago
• jiazhihao / sosp19ae
  Artifacts for SOSP'19 paper Optimizing Deep Learning Computation with Automatic Generation of Graph Substitutions
  ☆21Updated 3 years ago
• awslabs / ratex
  ☆23Updated 6 months ago
• mcrl / tccl
  Thunder Research Group's Collective Communication Library
  ☆37Updated last year
• MegEngine / cutlass-bak
  modified cutlass
  ☆14Updated 4 years ago
• microsoft / FractalTensor
  FractalTensor is a programming framework that introduces a novel approach to organizing data in deep neural networks (DNNs) as a list of …
  ☆25Updated 5 months ago
• Ascend / triton-ascend
  ☆48Updated this week
• K-Wu / pytorch-direct_dgl
  PyTorch-Direct code on top of PyTorch-1.8.0nightly (e152ca5) for Large Graph Convolutional Network Training with GPU-Oriented Data Commun…
  ☆44Updated last year
• feifeibear / PSTensor
  PSTensor provides a way to hack the memory management of tensors in TensorFlow and PyTorch by defining your own C++ Tensor Class.
  ☆10Updated 3 years ago
• nox-410 / tvm.tl
  An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.
  ☆50Updated 10 months ago
• apuaaChen / EVT_AE
  Artifacts of EVT ASPLOS'24
  ☆25Updated last year
• google / iopddl
  Supplemental materials for The ASPLOS 2025 / EuroSys 2025 Contest on Intra-Operator Parallelism for Distributed Deep Learning
  ☆23Updated 3 weeks ago
• microsoft / dist-ir
  An IR for efficiently simulating distributed ML computation.
  ☆28Updated last year
• iree-org / iree-nvgpu
  ☆50Updated last year
• comaniac / epoi
  Benchmark PyTorch Custom Operators
  ☆14Updated last year
• NVIDIA / HMM_sample_code
  CUDA 12.2 HMM demos
  ☆19Updated 10 months ago
• casys-kaist / EnvPipe
  ☆25Updated last year
• pigirons / conv3x3_m1
  This is a demo how to write a high performance convolution run on apple silicon
  ☆54Updated 3 years ago
• codyjrivera / tsm2x-imp
  Implementation of TSM2L and TSM2R -- High-Performance Tall-and-Skinny Matrix-Matrix Multiplication Algorithms for CUDA
  ☆32Updated 4 years ago