li-js / gpu_memory_profiling

Related projects: ⓘ

• adityaiitb / pyprof2
  PyProf2: PyTorch Profiling tool
  ☆83Updated 4 years ago
• microsoft / infinibatch
  Efficient, check-pointed data loading for deep learning with massive data sets.
  ☆203Updated last year
• harvardnlp / genbmm
  CUDA kernels for generalized matrix-multiplication in PyTorch
  ☆79Updated 2 years ago
• ptillet / torch-blocksparse
  Block-sparse primitives for PyTorch
  ☆147Updated 3 years ago
• HazyResearch / structured-nets
  Structured matrices for compressing neural networks
  ☆65Updated 11 months ago
• layer6ai-labs / T-Fixup
  Code for the ICML'20 paper "Improving Transformer Optimization Through Better Initialization"
  ☆89Updated 3 years ago
• Stonesjtu / pytorch-learning
  learning notes when learning the source code of pytorch
  ☆24Updated 5 years ago
• asappresearch / flop
  Pytorch library for factorized L0-based pruning.
  ☆42Updated 11 months ago
• quentinf00 / article-memory-log
  Implémentation of the article **Deep Learning CUDA Memory Usage and Pytorch optimization tricks**
  ☆42Updated 4 years ago
• TezRomacH / layer-to-layer-pytorch
  PyTorch implementation of L2L execution algorithm
  ☆107Updated last year
• tnq177 / transformers_without_tears
  Transformers without Tears: Improving the Normalization of Self-Attention
  ☆128Updated 3 months ago
• Separius / CudaRelativeAttention
  custom cuda kernel for {2, 3}d relative attention with pytorch wrapper
  ☆43Updated 4 years ago
• rosinality / adaptive-softmax-pytorch
  Adaptive Softmax implementation for PyTorch
  ☆79Updated 5 years ago
• gonglinyuan / StackingBERT
  Source code for "Efficient Training of BERT by Progressively Stacking"
  ☆111Updated 5 years ago
• khakhulin / compressed-transformer
  Compression of NMT transformer model with tensor methods
  ☆46Updated 5 years ago
• csrhddlam / pytorch-checkpoint
  ☆164Updated 5 years ago
• clovaai / length-adaptive-transformer
  Official Pytorch Implementation of Length-Adaptive Transformer (ACL 2021)
  ☆100Updated 3 years ago
• hendrycks / GELUs
  A smoother activation function (undergrad code)
  ☆104Updated 4 years ago
• cybertronai / transformer-xl
  Training Transformer-XL on 128 GPUs
  ☆140Updated 4 years ago
• erikwijmans / skynet-ddp-slurm-example
  Example of using PyTorch DistributedDataParallel and SLURM on skynet
  ☆29Updated 3 years ago
• rdspring1 / PyTorch_GBW_LM
  PyTorch Language Model for 1-Billion Word (LM1B / GBW) Dataset
  ☆122Updated 5 years ago
• allenai / sledgehammer
  ☆47Updated 4 years ago
• mrahtz / humble-gumbel
  Jupyter notebook on Gumbel-max and Gumbel-softmax tricks
  ☆39Updated last year
• wouterkool / stochastic-beam-search
  Implementation of Stochastic Beam Search using Fairseq
  ☆96Updated 5 years ago
• ofirpress / sandwich_transformer
  This repository contains the code for running the character-level Sandwich Transformers from our ACL 2020 paper on Improving Transformer …
  ☆55Updated 3 years ago
• huggingface / block_movement_pruning
  Block Sparse movement pruning
  ☆77Updated 3 years ago
• tunz / tcop-pytorch
  tunz's CUDA pytorch operator (MaskedSoftmax)
  ☆74Updated 5 years ago
• PetrochukM / HParams
  Configure Python functions explicitly and safely
  ☆126Updated last year
• toshas / torch-discounted-cumsum
  Fast Discounted Cumulative Sums in PyTorch
  ☆95Updated 3 years ago
• harvardnlp / cascaded-generation
  Cascaded Text Generation with Markov Transformers
  ☆126Updated last year