Alternatives and similar repositories for PSTensor

Users that are interested in PSTensor are comparing it to the libraries listed below

• Harry-Chen / InfMoE
  Inference framework for MoE layers based on TensorRT with Python binding
  ☆41Updated 4 years ago
• awslabs / lorien
  ☆42Updated 2 years ago
• chips-compilers-mlsys-21 / chips-compilers-mlsys-21.github.io
  ☆11Updated 4 years ago
• bytedance / QSync
  Official resporitory for "IPDPS' 24 QSync: Quantization-Minimized Synchronous Distributed Training Across Hybrid Devices".
  ☆20Updated last year
• zhuzilin / pytorch-malloc
  An external memory allocator example for PyTorch.
  ☆16Updated 3 months ago
• thu-pacman / PET
  PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections
  ☆121Updated 3 years ago
• uwsampl / dtr-prototype
  Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616
  ☆132Updated 2 years ago
• BaguaSys / bagua-net
  High performance NCCL plugin for Bagua.
  ☆15Updated 4 years ago
• zhuohan123 / terapipe
  ☆77Updated 4 years ago
• LeiWang1999 / Stream-k.tvm
  ☆19Updated 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 / tokenweave
  Efficient Compute-Communication Overlap for Distributed LLM Inference
  ☆62Updated 2 weeks ago
• saareliad / FTPipe
  FTPipe and related pipeline model parallelism research.
  ☆43Updated 2 years ago
• xuqifan897 / Optimus
  ☆28Updated 4 years ago
• jiazhihao / attention_superoptimizer
  An Attention Superoptimizer
  ☆22Updated 9 months ago
• byteps / examples
  BytePS examples (Vision, NLP, GAN, etc)
  ☆19Updated 2 years ago
• Ascend / torchair
  ☆19Updated this week
• awslabs / raf
  ☆145Updated 9 months ago
• DachengLi1 / AMP
  (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.
  ☆43Updated 3 years ago
• LeiWang1999 / AutoGPTQ.tvm
  GPTQ inference TVM kernel
  ☆39Updated last year
• Hsword / Hetu
  A high-performance distributed deep learning system targeting large-scale and automated distributed training. If you have any interests, …
  ☆123Updated last year
• cmu-catalyst / collage
  System for automated integration of deep learning backends.
  ☆47Updated 3 years ago
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆121Updated 11 months ago
• tile-ai / TileOPs
  ☆56Updated last week
• SJTU-IPADS / disb
  DISB is a new DNN inference serving benchmark with diverse workloads and models, as well as real-world traces.
  ☆54Updated last year
• nox-410 / tvm.tl
  An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.
  ☆51Updated last year
• lzhangbv / dear_pytorch
  [ICDCS 2023] DeAR: Accelerating Distributed Deep Learning with Fine-Grained All-Reduce Pipelining
  ☆11Updated last year
• jasperzhong / read-papers-and-code
  My paper/code reading notes in Chinese
  ☆46Updated 5 months 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 …
  ☆29Updated 10 months ago
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆137Updated 3 years ago