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
• 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
• Harry-Chen / InfMoE
  Inference framework for MoE layers based on TensorRT with Python binding
  ☆41Updated 4 years ago
• BaguaSys / bagua-net
  High performance NCCL plugin for Bagua.
  ☆15Updated 4 years ago
• zhuzilin / pytorch-malloc
  An external memory allocator example for PyTorch.
  ☆16Updated 4 months ago
• jiazhihao / attention_superoptimizer
  An Attention Superoptimizer
  ☆22Updated 11 months ago
• byteps / examples
  BytePS examples (Vision, NLP, GAN, etc)
  ☆19Updated 3 years ago
• uwsampl / dtr-prototype
  Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616
  ☆133Updated 2 years ago
• jasperzhong / read-papers-and-code
  My paper/code reading notes in Chinese
  ☆46Updated 6 months ago
• thu-pacman / PET
  PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections
  ☆122Updated 3 years 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
• awslabs / raf
  ☆145Updated 10 months ago
• awslabs / ratex
  ☆23Updated 4 months ago
• microsoft / tokenweave
  Efficient Compute-Communication Overlap for Distributed LLM Inference
  ☆66Updated last month
• comaniac / epoi
  Benchmark PyTorch Custom Operators
  ☆14Updated 2 years ago
• yuyangJin / PerFlow-AI
  PerFlow-AI is a programmable performance analysis, modeling, prediction tool for AI system.
  ☆25Updated last week
• geoffxy / habitat
  🔮 Execution time predictions for deep neural network training iterations across different GPUs.
  ☆63Updated 3 years ago
• zhuohan123 / terapipe
  ☆77Updated 4 years ago
• casys-kaist / EnvPipe
  ☆25Updated 2 years ago
• lzhangbv / dear_pytorch
  [ICDCS 2023] DeAR: Accelerating Distributed Deep Learning with Fine-Grained All-Reduce Pipelining
  ☆12Updated 2 years ago
• DachengLi1 / AMP
  (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.
  ☆43Updated 3 years ago
• Azure / msccl-executor-nccl
  ☆47Updated last year
• nox-410 / tvm.tl
  An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.
  ☆51Updated 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
• SJTU-IPADS / disb
  DISB is a new DNN inference serving benchmark with diverse workloads and models, as well as real-world traces.
  ☆57Updated last year
• xuqifan897 / Optimus
  ☆28Updated 4 years ago
• microsoft / dist-ir
  An IR for efficiently simulating distributed ML computation.
  ☆32Updated last year
• saareliad / FTPipe
  FTPipe and related pipeline model parallelism research.
  ☆43Updated 2 years ago
• Funatiq / gossip
  gossip: Efficient Communication Primitives for Multi-GPU Systems
  ☆60Updated 3 years ago