li-js / gpu_memory_profiling
Profile the GPU memory usage of every line in a Pytorch code
☆82Updated 6 years ago
Alternatives and similar repositories for gpu_memory_profiling:
Users that are interested in gpu_memory_profiling are comparing it to the libraries listed below
- Code for the ICML'20 paper "Improving Transformer Optimization Through Better Initialization"☆88Updated 4 years ago
- Structured matrices for compressing neural networks☆66Updated last year
- PyProf2: PyTorch Profiling tool☆82Updated 4 years ago
- CUDA kernels for generalized matrix-multiplication in PyTorch☆79Updated 3 years ago
- Pytorch library for factorized L0-based pruning.☆44Updated last year
- Efficient, check-pointed data loading for deep learning with massive data sets.☆205Updated last year
- ☆64Updated 4 years ago
- Code for paper "SWALP: Stochastic Weight Averaging forLow-Precision Training".☆62Updated 5 years ago
- ☆165Updated 6 years ago
- Adaptive Softmax implementation for PyTorch☆80Updated 5 years ago
- Fast Discounted Cumulative Sums in PyTorch☆95Updated 3 years ago
- Training Transformer-XL on 128 GPUs☆140Updated 4 years ago
- The official repository for our paper "The Devil is in the Detail: Simple Tricks Improve Systematic Generalization of Transformers". We s…☆67Updated 2 years ago
- Official Pytorch Implementation of Length-Adaptive Transformer (ACL 2021)☆101Updated 4 years ago
- Block-sparse primitives for PyTorch☆154Updated 3 years ago
- Compression of NMT transformer model with tensor methods☆48Updated 5 years ago
- ☆47Updated 4 years ago
- Reparameterize your PyTorch modules☆70Updated 4 years ago
- ☆62Updated 4 years ago
- meProp: Sparsified Back Propagation for Accelerated Deep Learning (ICML 2017)☆110Updated 2 years ago
- Explore OpenReview papers for ICLR (with citation counts, acceptance status, and ratings)!☆86Updated 3 years ago
- Customized matrix multiplication kernels☆53Updated 3 years ago
- Block Sparse movement pruning☆79Updated 4 years ago
- (Batched) advanced indexing for PyTorch.☆53Updated 3 months ago
- Configure Python functions explicitly and safely☆126Updated 4 months ago
- Discovering Neural Wirings (https://arxiv.org/abs/1906.00586)☆137Updated 5 years ago
- General stats about NeurIPS 2019☆28Updated 4 years ago
- An implementation of shampoo☆74Updated 7 years ago
- Code for SegTree Transformer (ICLR-RLGM 2019).☆27Updated 5 years ago
- [JMLR'20] NeurIPS 2019 MicroNet Challenge Efficient Language Modeling, Champion☆40Updated 4 years ago