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
- Efficient, check-pointed data loading for deep learning with massive data sets.☆205Updated last year
- PyProf2: PyTorch Profiling tool☆82Updated 4 years ago
- Customized matrix multiplication kernels☆53Updated 2 years ago
- Block-sparse primitives for PyTorch☆154Updated 3 years ago
- Pytorch library for factorized L0-based pruning.☆44Updated last year
- Adaptive Softmax implementation for PyTorch☆79Updated 5 years ago
- Official Pytorch Implementation of Length-Adaptive Transformer (ACL 2021)☆100Updated 4 years ago
- Source code for "Efficient Training of BERT by Progressively Stacking"☆113Updated 5 years ago
- Training Transformer-XL on 128 GPUs☆140Updated 4 years ago
- tunz's CUDA pytorch operator (MaskedSoftmax)☆75Updated 5 years ago
- Code for the ICML'20 paper "Improving Transformer Optimization Through Better Initialization"☆89Updated 3 years ago
- Block Sparse movement pruning☆78Updated 4 years ago
- Fast Discounted Cumulative Sums in PyTorch☆95Updated 3 years ago
- CUDA kernels for generalized matrix-multiplication in PyTorch☆79Updated 3 years ago
- Compression of NMT transformer model with tensor methods☆48Updated 5 years ago
- ☆47Updated 4 years ago
- A Toolkit for Training, Tracking, Saving Models and Syncing Results☆60Updated 4 years ago
- Implementation of Stochastic Beam Search using Fairseq☆99Updated 5 years ago
- ☆165Updated 6 years ago
- Transformers without Tears: Improving the Normalization of Self-Attention☆130Updated 7 months ago
- [JMLR'20] NeurIPS 2019 MicroNet Challenge Efficient Language Modeling, Champion☆40Updated 3 years ago
- A smoother activation function (undergrad code)☆108Updated 4 years ago
- Cascaded Text Generation with Markov Transformers☆129Updated last year
- Example of using PyTorch DistributedDataParallel and SLURM on skynet☆30Updated 3 years ago
- pytorch-profiler☆50Updated last year
- Structured matrices for compressing neural networks☆67Updated last year
- Code for SegTree Transformer (ICLR-RLGM 2019).☆27Updated 5 years ago
- Torch Distributed Experimental☆115Updated 5 months ago
- Official code for "Writing Distributed Applications with PyTorch", PyTorch Tutorial☆256Updated 2 years ago
- PyTorch implementation of L2L execution algorithm☆107Updated 2 years ago