quentinf00 / article-memory-log
Implémentation of the article **Deep Learning CUDA Memory Usage and Pytorch optimization tricks**
☆43Updated 5 years ago
Alternatives and similar repositories for article-memory-log:
Users that are interested in article-memory-log are comparing it to the libraries listed below
- ☆143Updated last year
- ☆70Updated 4 years ago
- Code for "Supermasks in Superposition"☆121Updated last year
- [NeurIPS 2022] DataMUX: Data Multiplexing for Neural Networks☆60Updated 2 years ago
- PyTorch implementation of HashedNets☆36Updated last year
- A custom PyTorch layer that is capable of implementing extremely wide and sparse linear layers efficiently☆49Updated last year
- Reproduction and analysis of SNIP paper☆30Updated 5 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆122Updated 6 months ago
- Block Sparse movement pruning☆78Updated 4 years ago
- Pytorch library for factorized L0-based pruning.☆44Updated last year
- ☆199Updated last year
- Soft Threshold Weight Reparameterization for Learnable Sparsity☆88Updated last year
- Block-sparse primitives for PyTorch☆154Updated 3 years ago
- [JMLR'20] NeurIPS 2019 MicroNet Challenge Efficient Language Modeling, Champion☆40Updated 3 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆84Updated last year
- Identify a binary weight or binary weight and activation subnetwork within a randomly initialized network by only pruning and binarizing …☆49Updated 2 years ago
- Study on the applicability of Direct Feedback Alignment to neural view synthesis, recommender systems, geometric learning, and natural la…☆86Updated 2 years ago
- Code accompanying the NeurIPS 2020 paper: WoodFisher (Singh & Alistarh, 2020)☆48Updated 3 years ago
- Official Pytorch Implementation of Length-Adaptive Transformer (ACL 2021)☆100Updated 4 years ago
- PyTorch implementation of L2L execution algorithm☆107Updated 2 years ago
- ☆74Updated 5 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆77Updated 4 years ago
- ☆222Updated 5 months ago
- Profile the GPU memory usage of every line in a Pytorch code☆82Updated 6 years ago
- ☆134Updated 3 years ago
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆58Updated 3 years ago
- Code for "Training Neural Networks with Fixed Sparse Masks" (NeurIPS 2021).☆58Updated 3 years ago
- Parameter Efficient Transfer Learning with Diff Pruning☆73Updated 3 years ago
- ☆188Updated 4 years ago
- Pretrained TorchVision models on CIFAR10 dataset (with weights)☆24Updated 4 years ago