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.☆208Updated 2 years ago
- Adaptive Softmax implementation for PyTorch☆81Updated 6 years ago
- Code for SegTree Transformer (ICLR-RLGM 2019).☆27Updated 5 years ago
- ☆165Updated 6 years ago
- CUDA kernels for generalized matrix-multiplication in PyTorch☆84Updated 3 years ago
- Alex Graves' Adaptive Computation Time in PyTorch☆14Updated 7 years ago
- PyTorch Language Model for 1-Billion Word (LM1B / GBW) Dataset☆123Updated 5 years ago
- Training Transformer-XL on 128 GPUs☆140Updated 5 years ago
- Source code for "Efficient Training of BERT by Progressively Stacking"☆112Updated 5 years ago
- Transformers without Tears: Improving the Normalization of Self-Attention☆131Updated last year
- Re-implementation of the Noise Contrastive Estimation algorithm for pyTorch, following "Noise-contrastive estimation: A new estimation pr…☆45Updated 5 years ago
- ☆64Updated 5 years ago
- Code for the ICML'20 paper "Improving Transformer Optimization Through Better Initialization"☆89Updated 4 years ago
- Fast Discounted Cumulative Sums in PyTorch☆96Updated 3 years ago
- Implements pytorch code for the Accelerated SGD algorithm.☆215Updated 7 years ago
- Reparameterize your PyTorch modules☆71Updated 4 years ago
- Train ImageNet in 18 minutes on AWS☆130Updated last year
- ☆47Updated 5 years ago
- Fork of diux-dev/imagenet18☆16Updated 6 years ago
- custom cuda kernel for {2, 3}d relative attention with pytorch wrapper☆43Updated 5 years ago
- Pytorch library for factorized L0-based pruning.☆45Updated last year
- Structured matrices for compressing neural networks☆67Updated last year
- Compression of NMT transformer model with tensor methods☆48Updated 6 years ago
- Experimental ground for optimizing memory of pytorch models☆366Updated 7 years ago
- meProp: Sparsified Back Propagation for Accelerated Deep Learning (ICML 2017)☆110Updated 3 years ago
- Official Pytorch Implementation of Length-Adaptive Transformer (ACL 2021)☆101Updated 4 years ago
- Training neural networks in TensorFlow 2.0 with 5x less memory☆132Updated 3 years ago
- learning notes when learning the source code of pytorch☆24Updated 6 years ago
- RNN benchmarks of pytorch, tensorflow and theano☆89Updated 6 years ago