IBM / pytorch-large-model-support

Related projects ⓘ

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

• IBM / tensorflow-large-model-support
  Large Model Support in Tensorflow
  ☆202Updated 4 years ago
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆495Updated 3 years ago
• ryujaehun / pytorch-gpu-benchmark
  Using the famous cnn model in Pytorch, we run benchmarks on various gpu.
  ☆227Updated 4 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 …
  ☆63Updated 2 years ago
• utsaslab / MONeT
  MONeT framework for reducing memory consumption of DNN training
  ☆173Updated 3 years ago
• uwsampl / dtr-prototype
  Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616
  ☆129Updated last year
• adityaiitb / pyprof2
  PyProf2: PyTorch Profiling tool
  ☆83Updated 4 years ago
• noahgolmant / pytorch-lars
  "Layer-wise Adaptive Rate Scaling" in PyTorch
  ☆86Updated 3 years ago
• jansel / pytorch-jit-paritybench
  ☆36Updated 5 months ago
• mlcommons / training_results_v1.0
  This repository contains the results and code for the MLPerf™ Training v1.0 benchmark.
  ☆37Updated 8 months ago
• mtbrandy / pytorch
  Please visit https://github.com/IBM/pytorch-large-model-support for the latest information on PyTorch LMS.
  ☆21Updated 4 years ago
• kaiyuyue / torchshard
  Slicing a PyTorch Tensor Into Parallel Shards
  ☆296Updated 3 years ago
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆129Updated 2 years ago
• petuum / autodist
  Simple Distributed Deep Learning on TensorFlow
  ☆134Updated 2 years ago
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆60Updated 8 months ago
• fumihwh / onnx-pytorch
  A code generator from ONNX to PyTorch code
  ☆133Updated 2 years ago
• jianweif / OptimalGradCheckpointing
  ☆39Updated 3 years ago
• mlcommons / training_results_v0.7
  This repository contains the results and code for the MLPerf™ Training v0.7 benchmark.
  ☆56Updated last year
• adityaiitb / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆22Updated 2 years ago
• richardkxu / distributed-pytorch
  Distributed, mixed-precision training with PyTorch
  ☆89Updated 4 years ago
• TezRomacH / layer-to-layer-pytorch
  PyTorch implementation of L2L execution algorithm
  ☆106Updated last year
• ucbrise / actnn
  ActNN: Reducing Training Memory Footprint via 2-Bit Activation Compressed Training
  ☆201Updated last year
• Manuscrit / SelectiveBackPropagation
  Implementation of the paper: Selective_Backpropagation from paper Accelerating Deep Learning by Focusing on the Biggest Losers
  ☆14Updated 4 years ago
• pytorch / torchdistx
  Torch Distributed Experimental
  ☆116Updated 3 months ago
• Distributed-AI / PipeTransformer
  PipeTransformer: Automated Elastic Pipelining for Distributed Training of Large-scale Models. ICML 2021
  ☆55Updated 3 years ago
• ptillet / torch-blocksparse
  Block-sparse primitives for PyTorch
  ☆148Updated 3 years ago
• spcl / substation
  Research and development for optimizing transformers
  ☆125Updated 3 years ago
• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆53Updated 2 years ago
• mlcommons / training_policies
  Issues related to MLPerf™ training policies, including rules and suggested changes
  ☆93Updated last month
• lambdal / deeplearning-benchmark
  Benchmark Suite for Deep Learning
  ☆250Updated 3 weeks ago