yaroslavvb / memory_util
TensorFlow util for building memory usage timeline from LOG_MEMORY messages
☆64Updated 7 years ago
Alternatives and similar repositories for memory_util:
Users that are interested in memory_util are comparing it to the libraries listed below
- Efficient layer normalization GPU kernel for Tensorflow☆110Updated 7 years ago
- TensorFlow kernels for probing memory☆15Updated 8 years ago
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆38Updated 6 years ago
- ☆23Updated 8 years ago
- Example of backprop which uses constant memory☆43Updated 6 years ago
- Python bindings for pyNVML and psutil library over network☆51Updated last year
- ☆28Updated 7 years ago
- Direct C++ Interface to PyTorch☆80Updated 6 years ago
- ☆57Updated 6 years ago
- A rudimentary wrapper around the fast Maxwell kernels for GEMM and convolution operations provided by nervanagpu☆34Updated 9 years ago
- easy embeddable Torch7 networks☆35Updated 8 years ago
- The Operator Vectorization Library, or OVL, is a python productivity library for defining high performance custom operators for the Tenso…☆68Updated 8 years ago
- Distributed Learning by Pair-Wise Averaging☆53Updated 7 years ago
- Signal Processing Library for PyTorch☆38Updated 7 years ago
- Weight initialization schemes for PyTorch nn.Modules☆70Updated 8 years ago
- Reference caffe implementation of LSUV initialization☆112Updated 7 years ago
- Code for Attentive Recurrent Comparators☆57Updated 8 years ago
- ☆92Updated 7 years ago
- Sublinear memory optimization for deep learning, reduce GPU memory cost to train deeper nets☆28Updated 8 years ago
- Training deep neural networks with low precision multiplications☆63Updated 9 years ago
- Tensorflow implementation of SGD with Coupled Adaptive Batch Size (CABS)☆43Updated 8 years ago
- Prunable nn layers for pytorch.☆48Updated 6 years ago
- train on AWS☆75Updated 6 years ago
- Just-in-time Dynamic Batching with MXNet Gluon.☆52Updated 4 years ago
- A pytorch implementation of "Self-Normalizing Neural Networks" by Klambauer et al. (still beta)☆59Updated 7 years ago
- Code used to generate the results appearing in "Train longer, generalize better: closing the generalization gap in large batch training o…☆148Updated 7 years ago
- ☆36Updated 5 years ago
- OpenCL Inference Engine for pytorch☆51Updated 7 years ago
- Caffe code and prototxt files for the CNN Design Patterns paper☆56Updated 8 years ago
- Implementation of Deep Residual Learning / Residual Network for MSR paper http://arxiv.org/abs/1512.03385☆37Updated 9 years ago