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:
- [NeurIPS 2022] DataMUX: Data Multiplexing for Neural Networks☆60Updated 3 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated 2 years ago
- ☆145Updated 2 years ago
- [JMLR'20] NeurIPS 2019 MicroNet Challenge Efficient Language Modeling, Champion☆41Updated 4 years ago
- End-to-end training of sparse deep neural networks with little-to-no performance loss.☆335Updated 3 years ago
- PyTorch implementation of HashedNets☆38Updated 2 years ago
- ☆54Updated 5 years ago
- A case study of efficient training of large language models using commodity hardware.☆68Updated 3 years ago
- Simply Numpy implementation of the FAVOR+ attention mechanism, https://teddykoker.com/2020/11/performers/☆38Updated 5 years ago
- Butterfly matrix multiplication in PyTorch☆178Updated 2 years ago
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆184Updated 4 years ago
- PyTorch implementation of L2L execution algorithm☆109Updated 3 years ago
- Torch Distributed Experimental☆117Updated last year
- HetSeq: Distributed GPU Training on Heterogeneous Infrastructure☆106Updated 2 years ago
- Structured matrices for compressing neural networks☆67Updated 2 years ago
- Code for "Supermasks in Superposition"☆125Updated 2 years ago
- ☆222Updated 2 years ago
- A collection of metrics to profile a single deep learning model or compare two different deep learning models☆27Updated 2 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆129Updated last year
- Pytorch library for factorized L0-based pruning.☆45Updated 2 years ago
- This repository contains example code to build models on TPUs☆30Updated 2 years ago
- ☆58Updated 3 years ago
- a lightweight transformer library for PyTorch☆72Updated 4 years ago
- Official code for "Distributed Deep Learning in Open Collaborations" (NeurIPS 2021)☆117Updated 4 years ago
- A collection of code snippets for my PyTorch Lightning projects☆107Updated 5 years ago
- Implementation of a Transformer, but completely in Triton☆279Updated 3 years ago
- ☆133Updated 4 years ago
- A custom PyTorch layer that is capable of implementing extremely wide and sparse linear layers efficiently☆51Updated 2 years ago
- A research library for pytorch-based neural network pruning, compression, and more.☆162Updated 3 years ago
- Examples of using PyTorch hooks, as covered in my YouTube tutorial video.☆39Updated 2 years ago