quentinf00 / article-memory-logLinks
Implémentation of the article **Deep Learning CUDA Memory Usage and Pytorch optimization tricks**
☆43Updated 6 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
Sorting:
- ☆145Updated 2 years ago
- [JMLR'20] NeurIPS 2019 MicroNet Challenge Efficient Language Modeling, Champion☆41Updated 4 years ago
- PyTorch implementation of L2L execution algorithm☆109Updated 3 years ago
- Butterfly matrix multiplication in PyTorch☆178Updated 2 years ago
- Structured matrices for compressing neural networks☆67Updated 2 years ago
- End-to-end training of sparse deep neural networks with little-to-no performance loss.☆335Updated 3 years ago
- Pytorch library for factorized L0-based pruning.☆45Updated 2 years ago
- Block-sparse primitives for PyTorch☆158Updated 4 years ago
- [ICML 2020] code for "PowerNorm: Rethinking Batch Normalization in Transformers" https://arxiv.org/abs/2003.07845☆120Updated 4 years ago
- PyTorch implementation of HashedNets☆38Updated 2 years ago
- A custom PyTorch layer that is capable of implementing extremely wide and sparse linear layers efficiently☆51Updated 2 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆129Updated last year
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated 2 years ago
- a lightweight transformer library for PyTorch☆72Updated 4 years ago
- A case study of efficient training of large language models using commodity hardware.☆68Updated 3 years ago
- ☆222Updated 2 years ago
- ☆54Updated 5 years ago
- [NeurIPS 2022] DataMUX: Data Multiplexing for Neural Networks☆60Updated 3 years ago
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆184Updated 4 years ago
- ☆47Updated 5 years ago
- ☆58Updated 3 years ago
- Train ImageNet in 18 minutes on AWS☆134Updated last year
- Discovering Neural Wirings (https://arxiv.org/abs/1906.00586)☆136Updated this week
- A research library for pytorch-based neural network pruning, compression, and more.☆162Updated 3 years ago
- Torch Distributed Experimental☆117Updated last year
- Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.☆111Updated 4 years ago
- ☆69Updated 5 years ago
- A collection of metrics to profile a single deep learning model or compare two different deep learning models☆27Updated 2 years ago
- ☆22Updated 7 years ago
- Profile the GPU memory usage of every line in a Pytorch code☆83Updated 7 years ago