Alternatives and similar repositories for pytorch-direct

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

• Lin-Mao / DrGPUM
  A memory profiler for NVIDIA GPUs to explore memory inefficiencies in GPU-accelerated applications.
  ☆25Updated 8 months ago
• jiazhihao / attention_superoptimizer
  An Attention Superoptimizer
  ☆22Updated 5 months ago
• 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 / FractalTensor
  FractalTensor is a programming framework that introduces a novel approach to organizing data in deep neural networks (DNNs) as a list of …
  ☆27Updated 6 months ago
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆64Updated last year
• zhuzilin / pytorch-malloc
  An external memory allocator example for PyTorch.
  ☆14Updated 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.
  ☆90Updated this week
• NVIDIA / HMM_sample_code
  CUDA 12.2 HMM demos
  ☆19Updated 11 months ago
• LeiWang1999 / Stream-k.tvm
  ☆19Updated 9 months ago
• sjtu-epcc / Tacker
  Tacker: Tensor-CUDA Core Kernel Fusion for Improving the GPU Utilization while Ensuring QoS
  ☆27Updated 4 months ago
• comaniac / epoi
  Benchmark PyTorch Custom Operators
  ☆14Updated last year
• apuaaChen / EVT_AE
  Artifacts of EVT ASPLOS'24
  ☆26Updated last year
• bytedance / QSync
  Official resporitory for "IPDPS' 24 QSync: Quantization-Minimized Synchronous Distributed Training Across Hybrid Devices".
  ☆20Updated last year
• awslabs / ratex
  ☆23Updated 7 months ago
• YashasSamaga / ConvolutionBuildingBlocks
  GEMM and Winograd based convolutions using CUTLASS
  ☆26Updated 4 years ago
• 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 2 years ago
• zhisbug / Cavs
  Cavs: An Efficient Runtime System for Dynamic Neural Networks
  ☆14Updated 4 years ago
• chips-compilers-mlsys-21 / chips-compilers-mlsys-21.github.io
  ☆11Updated 4 years ago
• 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
• osayamenja / Kleos
  Complete GPU residency for ML.
  ☆25Updated 2 weeks ago
• TiledTensor / TiledBench
  Benchmark tests supporting the TiledCUDA library.
  ☆16Updated 7 months ago
• jiazhihao / sosp19ae
  Artifacts for SOSP'19 paper Optimizing Deep Learning Computation with Automatic Generation of Graph Substitutions
  ☆21Updated 3 years ago
• illinois-impact / klap
  A source-to-source compiler for optimizing CUDA dynamic parallelism by aggregating launches
  ☆15Updated 6 years ago
• kanonjz / paper
  Machine Learning System
  ☆14Updated 5 years ago
• nox-410 / tvm.tl
  An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.
  ☆50Updated 11 months ago
• TiledTensor / TiledKernel
  TiledKernel is a code generation library based on macro kernels and memory hierarchy graph data structure.
  ☆19Updated last year
• LeiWang1999 / AutoGPTQ.tvm
  GPTQ inference TVM kernel
  ☆40Updated last year
• Jokeren / GPA
  GPU Performance Advisor
  ☆65Updated 2 years ago
• iree-org / iree-nvgpu
  ☆50Updated last year
• tlc-pack / TLCBench
  Benchmark scripts for TVM
  ☆74Updated 3 years ago