Alternatives and similar repositories for gpu_memory_profiling:

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

• adityaiitb / pyprof2
  PyProf2: PyTorch Profiling tool
  ☆82Updated 4 years ago
• ptillet / torch-blocksparse
  Block-sparse primitives for PyTorch
  ☆153Updated 3 years ago
• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆53Updated 2 years ago
• cybertronai / transformer-xl
  Training Transformer-XL on 128 GPUs
  ☆140Updated 4 years ago
• rosinality / adaptive-softmax-pytorch
  Adaptive Softmax implementation for PyTorch
  ☆80Updated 5 years ago
• waitwaitforget / modelparallel_pytorch
  Model Parallelism for pytorch training multiple network on multiple GPUs.
  ☆28Updated 6 years ago
• clovaai / length-adaptive-transformer
  Official Pytorch Implementation of Length-Adaptive Transformer (ACL 2021)
  ☆101Updated 4 years ago
• hendrycks / GELUs
  A smoother activation function (undergrad code)
  ☆108Updated 4 years ago
• layer6ai-labs / T-Fixup
  Code for the ICML'20 paper "Improving Transformer Optimization Through Better Initialization"
  ☆89Updated 4 years ago
• VoVAllen / tf-dlpack
  DLPack for Tensorflow
  ☆36Updated 4 years ago
• microsoft / infinibatch
  Efficient, check-pointed data loading for deep learning with massive data sets.
  ☆205Updated last year
• harvardnlp / genbmm
  CUDA kernels for generalized matrix-multiplication in PyTorch
  ☆79Updated 3 years ago
• ssnl / PyTorch-Reparam-Module
  Reparameterize your PyTorch modules
  ☆70Updated 4 years ago
• TezRomacH / layer-to-layer-pytorch
  PyTorch implementation of L2L execution algorithm
  ☆107Updated 2 years ago
• HazyResearch / structured-nets
  Structured matrices for compressing neural networks
  ☆66Updated last year
• TimDettmers / transformer-xl
  ☆64Updated 4 years ago
• toshas / torch-discounted-cumsum
  Fast Discounted Cumulative Sums in PyTorch
  ☆95Updated 3 years ago
• johnhw / differentiable_sorting
  Differentiable bitonic sorting
  ☆138Updated 4 years ago
• csrhddlam / pytorch-checkpoint
  ☆165Updated 6 years ago
• tnq177 / transformers_without_tears
  Transformers without Tears: Improving the Normalization of Self-Attention
  ☆130Updated 8 months ago
• asappresearch / flop
  Pytorch library for factorized L0-based pruning.
  ☆44Updated last year
• LeviViana / torch_sampling
  Efficient reservoir sampling implementation for PyTorch
  ☆107Updated 3 years ago
• harvardnlp / cascaded-generation
  Cascaded Text Generation with Markov Transformers
  ☆129Updated last year
• rdspring1 / PyTorch_GBW_LM
  PyTorch Language Model for 1-Billion Word (LM1B / GBW) Dataset
  ☆122Updated 5 years ago
• allenai / sledgehammer
  ☆47Updated 4 years ago
• khakhulin / compressed-transformer
  Compression of NMT transformer model with tensor methods
  ☆48Updated 5 years ago
• szymonmaszke / torchdatasets
  PyTorch dataset extended with map, cache etc. (tensorflow.data like)
  ☆328Updated 2 years ago
• hongyi-zhang / Fixup
  A Re-implementation of Fixed-update Initialization
  ☆152Updated 5 years ago
• yzh119 / segtree-transformer-v0
  Code for SegTree Transformer (ICLR-RLGM 2019).
  ☆27Updated 5 years ago
• MathInf / toroidal
  a lightweight transformer library for PyTorch
  ☆71Updated 3 years ago