Alternatives and similar repositories for pytorch-malloc

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

• bytedance / QSync
  Official resporitory for "IPDPS' 24 QSync: Quantization-Minimized Synchronous Distributed Training Across Hybrid Devices".
  ☆20Updated last year
• LeiWang1999 / Stream-k.tvm
  ☆20Updated last year
• chips-compilers-mlsys-21 / chips-compilers-mlsys-21.github.io
  ☆11Updated 4 years ago
• jiazhihao / attention_superoptimizer
  An Attention Superoptimizer
  ☆22Updated 11 months ago
• sjtu-epcc / DVABatch
  ☆21Updated 3 years ago
• TiledTensor / TiledLower
  TiledLower is a Dataflow Analysis and Codegen Framework written in Rust.
  ☆14Updated last year
• comaniac / epoi
  Benchmark PyTorch Custom Operators
  ☆14Updated 2 years ago
• xxyux / SpInfer
  SpInfer: Leveraging Low-Level Sparsity for Efficient Large Language Model Inference on GPUs
  ☆59Updated 9 months ago
• LeiWang1999 / AutoGPTQ.tvm
  GPTQ inference TVM kernel
  ☆41Updated last year
• nox-410 / tvm.tl
  An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.
  ☆51Updated last year
• apuaaChen / EVT_AE
  Artifacts of EVT ASPLOS'24
  ☆28Updated last year
• flashinfer-ai / debug-print
  Debug print operator for cudagraph debugging
  ☆14Updated last year
• microsoft / FractalTensor
  FractalTensor is a programming framework that introduces a novel approach to organizing data in deep neural networks (DNNs) as a list of …
  ☆31Updated last year
• chhzh123 / ptc-tutorial
  PyTorch compilation tutorial covering TorchScript, torch.fx, and Slapo
  ☆17Updated 2 years ago
• jack-willturner / nas-as-program-transformation-exploration
  The code for our paper "Neural Architecture Search as Program Transformation Exploration"
  ☆16Updated 4 years ago
• humuyan / Korch
  ASPLOS'24: Optimal Kernel Orchestration for Tensor Programs with Korch
  ☆40Updated 9 months ago
• thu-pacman / PET
  PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections
  ☆122Updated 3 years ago
• google / iopddl
  Supplemental materials for The ASPLOS 2025 / EuroSys 2025 Contest on Intra-Operator Parallelism for Distributed Deep Learning
  ☆25Updated 8 months ago
• IST-DASLab / gemm-fp8
  High Performance FP8 GEMM Kernels for SM89 and later GPUs.
  ☆20Updated 11 months ago
• UofT-EcoSystem / DietCode
  DietCode Code Release
  ☆65Updated 3 years ago
• tile-ai / TileOPs
  ☆78Updated this week
• 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
• casys-kaist / EnvPipe
  ☆25Updated 2 years ago
• LeiWang1999 / tvm_gpu_gemm
  play gemm with tvm
  ☆93Updated 2 years ago
• feifeibear / ChituAttention
  Quantized Attention on GPU
  ☆44Updated last year
• flashinfer-ai / cutlass-viz
  ☆65Updated 8 months ago
• tlc-pack / cutlass_fpA_intB_gemm
  A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer
  ☆96Updated 4 months ago
• marsupialtail / gpu-sparsert
  ☆18Updated 5 years ago
• cmu-catalyst / collage
  System for automated integration of deep learning backends.
  ☆47Updated 3 years ago
• ceruleangu / Block-Sparse-Benchmark
  Benchmark for matrix multiplications between dense and block sparse (BSR) matrix in TVM, blocksparse (Gray et al.) and cuSparse.
  ☆23Updated 5 years ago