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:
- Efficient, check-pointed data loading for deep learning with massive data sets.☆208Updated 2 years ago
- PyProf2: PyTorch Profiling tool☆82Updated 5 years ago
- CUDA kernels for generalized matrix-multiplication in PyTorch☆85Updated 3 years ago
- Structured matrices for compressing neural networks☆67Updated last year
- A smoother activation function (undergrad code)☆111Updated 5 years ago
- Code for the ICML'20 paper "Improving Transformer Optimization Through Better Initialization"☆89Updated 4 years ago
- ☆153Updated 3 years ago
- Memory efficient MAML using gradient checkpointing☆85Updated 5 years ago
- Configure Python functions explicitly and safely☆126Updated 7 months ago
- Adaptive Softmax implementation for PyTorch☆81Updated 6 years ago
- ☆62Updated 5 years ago
- Training Transformer-XL on 128 GPUs☆140Updated 5 years ago
- Reparameterize your PyTorch modules☆71Updated 4 years ago
- ☆64Updated 5 years ago
- ☆165Updated 6 years ago
- Explore OpenReview papers for ICLR (with citation counts, acceptance status, and ratings)!☆88Updated 4 years ago
- Efficient reservoir sampling implementation for PyTorch☆106Updated 3 years ago
- Code for SegTree Transformer (ICLR-RLGM 2019).☆27Updated 5 years ago
- [ICML 2020] code for "PowerNorm: Rethinking Batch Normalization in Transformers" https://arxiv.org/abs/2003.07845☆120Updated 4 years ago
- This repository is no longer maintained. Check☆81Updated 5 years ago
- MTAdam: Automatic Balancing of Multiple Training Loss Terms☆36Updated 4 years ago
- ☆14Updated 6 years ago
- ☆33Updated 5 years ago
- Source code for "Efficient Training of BERT by Progressively Stacking"☆112Updated 6 years ago
- Differentiable bitonic sorting☆141Updated 5 years ago
- Fast Discounted Cumulative Sums in PyTorch☆96Updated 3 years ago
- tunz's CUDA pytorch operator (MaskedSoftmax)☆75Updated 6 years ago
- An implementation of shampoo☆75Updated 7 years ago
- SVG/Javascript-based library for creating presentations/figures☆59Updated 2 years ago
- Sparse and structured neural attention mechanisms☆224Updated 4 years ago