quentinf00 / article-memory-logLinks
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
Sorting:
- PyTorch implementation of L2L execution algorithm☆108Updated 2 years ago
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆182Updated 4 years ago
- ☆144Updated 2 years ago
- Pytorch library for factorized L0-based pruning.☆45Updated last year
- HetSeq: Distributed GPU Training on Heterogeneous Infrastructure☆106Updated 2 years ago
- [JMLR'20] NeurIPS 2019 MicroNet Challenge Efficient Language Modeling, Champion☆40Updated 4 years ago
- Implementation of a Transformer, but completely in Triton☆273Updated 3 years ago
- Collection of the latest, greatest, deep learning optimizers (for Pytorch) - CNN, NLP suitable☆216Updated 4 years ago
- A collection of code snippets for my PyTorch Lightning projects☆109Updated 4 years ago
- Structured matrices for compressing neural networks☆67Updated last year
- End-to-end training of sparse deep neural networks with little-to-no performance loss.☆324Updated 2 years ago
- a lightweight transformer library for PyTorch☆72Updated 3 years ago
- Block-sparse primitives for PyTorch☆158Updated 4 years ago
- Block Sparse movement pruning☆81Updated 4 years ago
- A case study of efficient training of large language models using commodity hardware.☆68Updated 3 years ago
- Official code for "Distributed Deep Learning in Open Collaborations" (NeurIPS 2021)☆117Updated 3 years ago
- [ICML 2020] code for "PowerNorm: Rethinking Batch Normalization in Transformers" https://arxiv.org/abs/2003.07845☆120Updated 4 years ago
- Butterfly matrix multiplication in PyTorch☆174Updated last year
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated last year
- A collection of metrics to profile a single deep learning model or compare two different deep learning models☆27Updated last year
- [NeurIPS 2022] DataMUX: Data Multiplexing for Neural Networks☆60Updated 2 years ago
- ☆57Updated 3 years ago
- This repository contains example code to build models on TPUs☆30Updated 2 years ago
- A research library for pytorch-based neural network pruning, compression, and more.☆162Updated 2 years ago
- DiffQ performs differentiable quantization using pseudo quantization noise. It can automatically tune the number of bits used per weight …☆236Updated 2 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆126Updated last year
- Simply Numpy implementation of the FAVOR+ attention mechanism, https://teddykoker.com/2020/11/performers/☆38Updated 4 years ago
- ☆133Updated 4 years ago
- ☆54Updated 4 years ago
- Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch☆184Updated 2 years ago