Alternatives and similar repositories for pytorch-large-model-support

Users that are interested in pytorch-large-model-support are comparing it to the libraries listed below

• ryujaehun / pytorch-gpu-benchmark
  Using the famous cnn model in Pytorch, we run benchmarks on various gpu.
  ☆242Updated last year
• utsaslab / MONeT
  MONeT framework for reducing memory consumption of DNN training
  ☆174Updated 4 years ago
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆504Updated 4 years ago
• adityaiitb / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆22Updated 3 years ago
• ucbrise / actnn
  ActNN: Reducing Training Memory Footprint via 2-Bit Activation Compressed Training
  ☆200Updated 2 years ago
• noahgolmant / pytorch-lars
  "Layer-wise Adaptive Rate Scaling" in PyTorch
  ☆87Updated 4 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
• adityaiitb / pyprof2
  PyProf2: PyTorch Profiling tool
  ☆82Updated 5 years ago
• IBM / tensorflow-large-model-support
  Large Model Support in Tensorflow
  ☆202Updated 4 years ago
• frgfm / torch-scan
  Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)
  ☆221Updated 5 months ago
• kaiyuyue / torchshard
  Slicing a PyTorch Tensor Into Parallel Shards
  ☆300Updated 2 months ago
• yaysummeriscoming / DALI_pytorch_demo
  Example code showing how to use Nvidia DALI in pytorch, with fallback to torchvision. Contains a few differences to the official Nvidia …
  ☆197Updated 5 years ago
• NVlabs / condensa
  Programmable Neural Network Compression
  ☆149Updated 3 years ago
• apple / ml-cifar-10-faster
  ☆57Updated 3 years ago
• jianweif / OptimalGradCheckpointing
  ☆41Updated 4 years ago
• snuspl / nimble
  Lightweight and Parallel Deep Learning Framework
  ☆264Updated 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
• fumihwh / onnx-pytorch
  A code generator from ONNX to PyTorch code
  ☆139Updated 2 years ago
• pytorch / torchdistx
  Torch Distributed Experimental
  ☆117Updated last year
• mlcommons / training_results_v1.0
  This repository contains the results and code for the MLPerf™ Training v1.0 benchmark.
  ☆37Updated last year
• LukasHedegaard / pytorch-benchmark
  Easily benchmark PyTorch model FLOPs, latency, throughput, allocated gpu memory and energy consumption
  ☆106Updated 2 years ago
• Distributed-AI / PipeTransformer
  PipeTransformer: Automated Elastic Pipelining for Distributed Training of Large-scale Models. ICML 2021
  ☆56Updated 4 years ago
• csrhddlam / pytorch-checkpoint
  ☆165Updated 6 years ago
• gpuopenanalytics / pynvml
  Provide Python access to the NVML library for GPU diagnostics
  ☆245Updated 8 months ago
• NVlabs / tensorcom
  ☆109Updated 4 years ago
• TezRomacH / layer-to-layer-pytorch
  PyTorch implementation of L2L execution algorithm
  ☆108Updated 2 years ago
• suvojit-0x55aa / mixed-precision-pytorch
  Training with FP16 weights in PyTorch
  ☆79Updated 6 years ago
• prigoyal / pytorch_memonger
  Experimental ground for optimizing memory of pytorch models
  ☆366Updated 7 years ago
• cli99 / flops-profiler
  pytorch-profiler
  ☆51Updated 2 years ago
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆132Updated 3 years ago