Alternatives and similar repositories for PSTensor

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

• awslabs / lorien
  ☆42Updated 2 years ago
• zhuzilin / pytorch-malloc
  An external memory allocator example for PyTorch.
  ☆16Updated last month
• Harry-Chen / InfMoE
  Inference framework for MoE layers based on TensorRT with Python binding
  ☆41Updated 4 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
• thu-pacman / PET
  PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections
  ☆122Updated 3 years ago
• jiazhihao / attention_superoptimizer
  An Attention Superoptimizer
  ☆22Updated 8 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
• zhuohan123 / terapipe
  ☆75Updated 4 years ago
• byteps / examples
  BytePS examples (Vision, NLP, GAN, etc)
  ☆19Updated 2 years ago
• Azure / msccl-executor-nccl
  ☆46Updated 9 months ago
• nox-410 / tvm.tl
  An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.
  ☆51Updated last year
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆135Updated 3 years ago
• awslabs / raf
  ☆145Updated 8 months ago
• Hsword / Hetu
  A high-performance distributed deep learning system targeting large-scale and automated distributed training. If you have any interests, …
  ☆121Updated last year
• saareliad / FTPipe
  FTPipe and related pipeline model parallelism research.
  ☆42Updated 2 years ago
• uwsampl / nexus
  ☆84Updated 3 months ago
• google / iopddl
  Supplemental materials for The ASPLOS 2025 / EuroSys 2025 Contest on Intra-Operator Parallelism for Distributed Deep Learning
  ☆23Updated 4 months ago
• awslabs / ratex
  ☆23Updated last month
• cmu-catalyst / collage
  System for automated integration of deep learning backends.
  ☆47Updated 3 years ago
• microsoft / tokenweave
  Efficient Compute-Communication Overlap for Distributed LLM Inference
  ☆58Updated this week
• KuangjuX / NVSHMEM-Tutorial
  NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer
  ☆130Updated 2 weeks ago
• octoml / synr
  A library for syntactically rewriting Python programs, pronounced (sinner).
  ☆68Updated 3 years ago
• 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
• DachengLi1 / AMP
  (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.
  ☆41Updated 2 years ago
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆121Updated 10 months ago
• jasperzhong / read-papers-and-code
  My paper/code reading notes in Chinese
  ☆46Updated 3 months ago
• alibaba / easydist
  Automated Parallelization System and Infrastructure for Multiple Ecosystems
  ☆80Updated 10 months ago
• tile-ai / TileOPs
  ☆51Updated last week