Alternatives and similar repositories for gpu_memory_profiling

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

• harvardnlp / genbmm
  CUDA kernels for generalized matrix-multiplication in PyTorch
  ☆79Updated 3 years ago
• adityaiitb / pyprof2
  PyProf2: PyTorch Profiling tool
  ☆82Updated 4 years ago
• rosinality / adaptive-softmax-pytorch
  Adaptive Softmax implementation for PyTorch
  ☆80Updated 6 years ago
• microsoft / infinibatch
  Efficient, check-pointed data loading for deep learning with massive data sets.
  ☆207Updated last year
• PetrochukM / HParams
  Configure Python functions explicitly and safely
  ☆126Updated 5 months ago
• csrhddlam / pytorch-checkpoint
  ☆165Updated 6 years ago
• ag14774 / diffdist
  ☆62Updated 4 years ago
• ptillet / torch-blocksparse
  Block-sparse primitives for PyTorch
  ☆155Updated 4 years ago
• ssnl / PyTorch-Reparam-Module
  Reparameterize your PyTorch modules
  ☆71Updated 4 years ago
• cybertronai / transformer-xl
  Training Transformer-XL on 128 GPUs
  ☆140Updated 4 years ago
• gonglinyuan / StackingBERT
  Source code for "Efficient Training of BERT by Progressively Stacking"
  ☆112Updated 5 years ago
• rahulkidambi / AccSGD
  Implements pytorch code for the Accelerated SGD algorithm.
  ☆215Updated 7 years ago
• rdspring1 / PyTorch_GBW_LM
  PyTorch Language Model for 1-Billion Word (LM1B / GBW) Dataset
  ☆123Updated 5 years ago
• LeviViana / torch_sampling
  Efficient reservoir sampling implementation for PyTorch
  ☆106Updated 3 years ago
• layer6ai-labs / T-Fixup
  Code for the ICML'20 paper "Improving Transformer Optimization Through Better Initialization"
  ☆88Updated 4 years ago
• VoVAllen / tf-dlpack
  DLPack for Tensorflow
  ☆35Updated 5 years ago
• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆54Updated 3 years ago
• tunz / tcop-pytorch
  tunz's CUDA pytorch operator (MaskedSoftmax)
  ☆75Updated 6 years ago
• toiaydcdyywlhzvlob / backpack
  This repository is no longer maintained. Check
  ☆81Updated 5 years ago
• toshas / torch-discounted-cumsum
  Fast Discounted Cumulative Sums in PyTorch
  ☆95Updated 3 years ago
• lancopku / meProp
  meProp: Sparsified Back Propagation for Accelerated Deep Learning (ICML 2017)
  ☆110Updated 3 years ago
• noahgolmant / pytorch-lars
  "Layer-wise Adaptive Rate Scaling" in PyTorch
  ☆86Updated 4 years ago
• oval-group / mlogger
  a lightweight and simple logger for Machine Learning
  ☆127Updated 4 years ago
• asappresearch / flop
  Pytorch library for factorized L0-based pruning.
  ☆45Updated last year
• clovaai / length-adaptive-transformer
  Official Pytorch Implementation of Length-Adaptive Transformer (ACL 2021)
  ☆101Updated 4 years ago
• TezRomacH / layer-to-layer-pytorch
  PyTorch implementation of L2L execution algorithm
  ☆107Updated 2 years ago
• cybertronai / imagenet18
  Train ImageNet in 18 minutes on AWS
  ☆129Updated last year
• HazyResearch / structured-nets
  Structured matrices for compressing neural networks
  ☆66Updated last year
• khakhulin / compressed-transformer
  Compression of NMT transformer model with tensor methods
  ☆48Updated 5 years ago
• allenai / sledgehammer
  ☆47Updated 4 years ago