yaroslavvb / chain_constant_memoryLinks
Example of backprop which uses constant memory
☆43Updated 6 years ago
Alternatives and similar repositories for chain_constant_memory
Users that are interested in chain_constant_memory are comparing it to the libraries listed below
Sorting:
- TensorFlow util for building memory usage timeline from LOG_MEMORY messages☆65Updated 7 years ago
- PyTorch development for onnx☆21Updated 7 years ago
- Code used to generate the results appearing in "Train longer, generalize better: closing the generalization gap in large batch training o…☆149Updated 8 years ago
- Simple example of implementing a new Tensorflow operation and its gradient in C++.☆56Updated 6 years ago
- ☆42Updated 5 years ago
- Training wide residual networks for deployment using a single bit for each weight - Official Code Repository for ICLR 2018 Published Pape…☆36Updated 5 years ago
- ☆37Updated 6 years ago
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆38Updated 6 years ago
- Tensorflow implementation of SGD with Coupled Adaptive Batch Size (CABS)☆43Updated 8 years ago
- Use TensorFlow efficiently☆95Updated 4 years ago
- Prunable nn layers for pytorch.☆48Updated 6 years ago
- Just-in-time Dynamic Batching with MXNet Gluon.☆52Updated 5 years ago
- Odds and Ends and Things I've implemented.☆78Updated 6 years ago
- Distributed Learning by Pair-Wise Averaging☆52Updated 7 years ago
- ☆57Updated 7 years ago
- Efficient layer normalization GPU kernel for Tensorflow☆111Updated 8 years ago
- ☆23Updated 9 years ago
- Lasagne code for weight normalization☆88Updated 9 years ago
- Implementation of "Learning with Random Learning Rates" in PyTorch.☆102Updated 5 years ago
- PyProf2: PyTorch Profiling tool☆82Updated 5 years ago
- Deep learning with a multiplication budget☆47Updated 6 years ago
- TensorFlow implementation of (Momentum) Stochastic Variance-Adapted Gradient.