li-js / gpu_memory_profiling

Related projects ⓘ

Alternatives and complementary repositories for gpu_memory_profiling

• adityaiitb / pyprof2
  PyProf2: PyTorch Profiling tool
  ☆83Updated 4 years ago
• microsoft / infinibatch
  Efficient, check-pointed data loading for deep learning with massive data sets.
  ☆205Updated last year
• rosinality / adaptive-softmax-pytorch
  Adaptive Softmax implementation for PyTorch
  ☆79Updated 5 years ago
• ptillet / torch-blocksparse
  Block-sparse primitives for PyTorch
  ☆148Updated 3 years ago
• clovaai / length-adaptive-transformer
  Official Pytorch Implementation of Length-Adaptive Transformer (ACL 2021)
  ☆100Updated 4 years ago
• harvardnlp / genbmm
  CUDA kernels for generalized matrix-multiplication in PyTorch
  ☆79Updated 3 years ago
• layer6ai-labs / T-Fixup
  Code for the ICML'20 paper "Improving Transformer Optimization Through Better Initialization"
  ☆89Updated 3 years ago
• TimDettmers / transformer-xl
  ☆64Updated 4 years ago
• gonglinyuan / StackingBERT
  Source code for "Efficient Training of BERT by Progressively Stacking"
  ☆112Updated 5 years ago
• asappresearch / flop
  Pytorch library for factorized L0-based pruning.
  ☆43Updated last year
• stevenygd / SWALP
  Code for paper "SWALP: Stochastic Weight Averaging forLow-Precision Training".
  ☆62Updated 5 years ago
• tnq177 / transformers_without_tears
  Transformers without Tears: Improving the Normalization of Self-Attention
  ☆130Updated 5 months ago
• HazyResearch / structured-nets
  Structured matrices for compressing neural networks
  ☆67Updated last year
• allenai / sledgehammer
  ☆47Updated 4 years ago
• toshas / torch-discounted-cumsum
  Fast Discounted Cumulative Sums in PyTorch
  ☆95Updated 3 years ago
• hendrycks / GELUs
  A smoother activation function (undergrad code)
  ☆108Updated 4 years ago
• LiyuanLucasLiu / Torch-Scope
  A Toolkit for Training, Tracking, Saving Models and Syncing Results
  ☆60Updated 4 years ago
• RobertCsordas / transformer_generalization
  The official repository for our paper "The Devil is in the Detail: Simple Tricks Improve Systematic Generalization of Transformers". We s…
  ☆66Updated last year
• tunz / tcop-pytorch
  tunz's CUDA pytorch operator (MaskedSoftmax)
  ☆75Updated 5 years ago
• khakhulin / compressed-transformer
  Compression of NMT transformer model with tensor methods
  ☆46Updated 5 years ago
• rdspring1 / PyTorch_GBW_LM
  PyTorch Language Model for 1-Billion Word (LM1B / GBW) Dataset
  ☆123Updated 5 years ago
• yzh119 / segtree-transformer-v0
  Code for SegTree Transformer (ICLR-RLGM 2019).
  ☆27Updated 5 years ago
• ag14774 / diffdist
  ☆61Updated 4 years ago
• huggingface / block_movement_pruning
  Block Sparse movement pruning
  ☆78Updated 3 years ago
• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆53Updated 2 years ago
• sIncerass / powernorm
  [ICML 2020] code for "PowerNorm: Rethinking Batch Normalization in Transformers" https://arxiv.org/abs/2003.07845
  ☆119Updated 3 years ago
• cybertronai / transformer-xl
  Training Transformer-XL on 128 GPUs
  ☆140Updated 4 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
• noahgolmant / pytorch-lars
  "Layer-wise Adaptive Rate Scaling" in PyTorch
  ☆86Updated 3 years ago
• harvardnlp / cascaded-generation
  Cascaded Text Generation with Markov Transformers
  ☆128Updated last year