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:
- ☆143Updated 2 years ago
- PyTorch implementation of L2L execution algorithm☆108Updated 2 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆82Updated 2 years ago
- End-to-end training of sparse deep neural networks with little-to-no performance loss.☆324Updated 2 years ago
- ☆54Updated 5 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
- [JMLR'20] NeurIPS 2019 MicroNet Challenge Efficient Language Modeling, Champion☆40Updated 4 years ago
- a lightweight transformer library for PyTorch☆72Updated 3 years ago
- A collection of metrics to profile a single deep learning model or compare two different deep learning models☆27Updated last year
- Butterfly matrix multiplication in PyTorch☆174Updated 2 years ago
- Block-sparse primitives for PyTorch☆160Updated 4 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆129Updated last year
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆184Updated 4 years ago
- Structured matrices for compressing neural networks☆67Updated 2 years ago
- Explores the ideas presented in Deep Ensembles: A Loss Landscape Perspective (https://arxiv.org/abs/1912.02757) by Stanislav Fort, Huiyi …☆66Updated 5 years ago
- HetSeq: Distributed GPU Training on Heterogeneous Infrastructure☆106Updated 2 years ago
- Torch Distributed Experimental☆117Updated last year
- ☆133Updated 4 years ago
- A custom PyTorch layer that is capable of implementing extremely wide and sparse linear layers efficiently☆51Updated last year
- ☆218Updated 2 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 4 years ago
- Block Sparse movement pruning☆81Updated 4 years ago
- Simply Numpy implementation of the FAVOR+ attention mechanism, https://teddykoker.com/2020/11/performers/☆38Updated 4 years ago
- ☆22Updated 7 years ago
- Implementation of a Transformer, but completely in Triton☆275Updated 3 years ago
- Code for "Supermasks in Superposition"☆124Updated 2 years ago
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆63Updated 4 years ago
- [ICLR 2020] Drawing Early-Bird Tickets: Toward More Efficient Training of Deep Networks☆139Updated 4 years ago