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
- Pytorch library for factorized L0-based pruning.☆44Updated last year
- PyTorch implementation of HashedNets☆36Updated last year
- Structured matrices for compressing neural networks☆66Updated last year
- ☆70Updated 4 years ago
- Code for "Supermasks in Superposition"☆121Updated last year
- Code accompanying the NeurIPS 2020 paper: WoodFisher (Singh & Alistarh, 2020)☆48Updated 3 years ago
- Block Sparse movement pruning☆78Updated 4 years ago
- ☆143Updated last year
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆59Updated 3 years ago
- Block-sparse primitives for PyTorch☆153Updated 3 years ago
- Code for "Training Neural Networks with Fixed Sparse Masks" (NeurIPS 2021).☆58Updated 3 years ago
- ☆22Updated 6 years ago
- [JMLR'20] NeurIPS 2019 MicroNet Challenge Efficient Language Modeling, Champion☆40Updated 3 years ago
- ☆201Updated 2 years ago
- [NeurIPS 2020] "The Lottery Ticket Hypothesis for Pre-trained BERT Networks", Tianlong Chen, Jonathan Frankle, Shiyu Chang, Sijia Liu, Ya…☆140Updated 3 years ago
- A custom PyTorch layer that is capable of implementing extremely wide and sparse linear layers efficiently☆49Updated last year
- This repository contains code to replicate the experiments given in NeurIPS 2019 paper "One ticket to win them all: generalizing lottery …☆51Updated 6 months ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆77Updated 4 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆123Updated 7 months ago
- PyTorch implementation of L2L execution algorithm☆107Updated 2 years ago
- Soft Threshold Weight Reparameterization for Learnable Sparsity☆87Updated 2 years ago
- ☆39Updated 5 years ago
- Parameter Efficient Transfer Learning with Diff Pruning☆73Updated 4 years ago
- Code for paper "SWALP: Stochastic Weight Averaging forLow-Precision Training".☆62Updated 5 years ago
- ☆222Updated 6 months ago
- [ICLR 2023] "Sparsity May Cry: Let Us Fail (Current) Sparse Neural Networks Together!" Shiwei Liu, Tianlong Chen, Zhenyu Zhang, Xuxi Chen…☆27Updated last year
- ☆19Updated 2 years ago
- Reproduction and analysis of SNIP paper☆30Updated 5 years ago
- A research library for pytorch-based neural network pruning, compression, and more.☆160Updated 2 years ago
- ☆82Updated 5 years ago