Alternatives and similar repositories for PSTensor

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

• awslabs / lorien
  ☆43Updated 2 years ago
• uwsampl / dtr-prototype
  Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616
  ☆132Updated 2 years ago
• zhuzilin / pytorch-malloc
  An external memory allocator example for PyTorch.
  ☆16Updated last month
• jasperzhong / read-papers-and-code
  My paper/code reading notes in Chinese
  ☆46Updated 3 months ago
• 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
• bytedance / QSync
  Official resporitory for "IPDPS' 24 QSync: Quantization-Minimized Synchronous Distributed Training Across Hybrid Devices".
  ☆20Updated last year
• byteps / examples
  BytePS examples (Vision, NLP, GAN, etc)
  ☆19Updated 2 years ago
• thu-pacman / PET
  PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections
  ☆122Updated 3 years ago
• zhuohan123 / terapipe
  ☆75Updated 4 years ago
• saareliad / FTPipe
  FTPipe and related pipeline model parallelism research.
  ☆42Updated 2 years ago
• netx-repo / PipeSwitch
  PipeSwitch: Fast Pipelined Context Switching for Deep Learning Applications
  ☆127Updated 3 years ago
• jiazhihao / attention_superoptimizer
  An Attention Superoptimizer
  ☆22Updated 7 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
• DachengLi1 / AMP
  (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.
  ☆40Updated 2 years ago
• SJTU-IPADS / disb
  DISB is a new DNN inference serving benchmark with diverse workloads and models, as well as real-world traces.
  ☆53Updated last year
• lzhangbv / dear_pytorch
  [ICDCS 2023] DeAR: Accelerating Distributed Deep Learning with Fine-Grained All-Reduce Pipelining
  ☆12Updated last year
• sjtu-epcc / Tacker
  Tacker: Tensor-CUDA Core Kernel Fusion for Improving the GPU Utilization while Ensuring QoS
  ☆31Updated 7 months ago
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆134Updated 3 years ago
• dywsjtu / apparate
  Artifact for "Apparate: Rethinking Early Exits to Tame Latency-Throughput Tensions in ML Serving" [SOSP '24]
  ☆25Updated 9 months ago
• uwsampl / nexus
  ☆84Updated 2 months ago
• casys-kaist / EnvPipe
  ☆25Updated 2 years ago
• geoffxy / habitat
  🔮 Execution time predictions for deep neural network training iterations across different GPUs.
  ☆63Updated 2 years ago
• octoml / synr
  A library for syntactically rewriting Python programs, pronounced (sinner).
  ☆69Updated 3 years ago
• nox-410 / tvm.tl
  An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.
  ☆51Updated last year
• stanford-mast / INFaaS
  Model-less Inference Serving
  ☆92Updated last year
• comaniac / epoi
  Benchmark PyTorch Custom Operators
  ☆14Updated 2 years 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
• awslabs / raf
  ☆144Updated 7 months ago
• sjtu-epcc / DVABatch
  ☆20Updated 3 years ago