li-js / gpu_memory_profilingLinks
Profile the GPU memory usage of every line in a Pytorch code
☆83Updated 7 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
Sorting:
- PyProf2: PyTorch Profiling tool☆82Updated 5 years ago
- Efficient, check-pointed data loading for deep learning with massive data sets.☆210Updated 2 years ago
- CUDA kernels for generalized matrix-multiplication in PyTorch☆85Updated 4 years ago
- ☆62Updated 5 years ago
- Reparameterize your PyTorch modules☆71Updated 4 years ago
- Differentiable bitonic sorting☆141Updated 5 years ago
- Configure Python functions explicitly and safely☆127Updated last year
- Structured matrices for compressing neural networks☆67Updated 2 years ago
- ☆165Updated 6 years ago
- Efficient reservoir sampling implementation for PyTorch☆107Updated 4 years ago
- This repository is no longer maintained. Check☆81Updated 5 years ago
- meProp: Sparsified Back Propagation for Accelerated Deep Learning (ICML 2017)☆110Updated 3 years ago
- ☆14Updated 6 years ago
- Memory efficient MAML using gradient checkpointing☆86Updated 5 years ago
- Code for paper "SWALP: Stochastic Weight Averaging forLow-Precision Training".☆62Updated 6 years ago
- Jupyter notebook on Gumbel-max and Gumbel-softmax tricks☆40Updated 3 years ago
- Code for SegTree Transformer (ICLR-RLGM 2019).☆27Updated 6 years ago
- Adaptive Softmax implementation for PyTorch☆81Updated 6 years ago
- Training Transformer-XL on 128 GPUs☆141Updated 5 years ago
- Code for the ICML'20 paper "Improving Transformer Optimization Through Better Initialization"☆89Updated 4 years ago
- Train ImageNet in 18 minutes on AWS☆134Updated last year
- Block-sparse primitives for PyTorch☆160Updated 4 years ago
- ☆65Updated 5 years ago
- Make quick mechanical turk HTML/Javascript interfaces and launch them with Python functions☆41Updated 4 years ago
- PyTorch implementation of L2L execution algorithm☆109Updated 2 years ago
- Fast Discounted Cumulative Sums in PyTorch☆96Updated 4 years ago
- Discovering Neural Wirings (https://arxiv.org/abs/1906.00586)☆136Updated 5 years ago
- Implements pytorch code for the Accelerated SGD algorithm.☆215Updated 7 years ago
- A Re-implementation of Fixed-update Initialization☆155Updated 6 years ago
- [JMLR'20] NeurIPS 2019 MicroNet Challenge Efficient Language Modeling, Champion☆41Updated 4 years ago