yaroslavvb / chain_constant_memory
Example of backprop which uses constant memory
☆43Updated 6 years ago
Related projects ⓘ
Alternatives and complementary repositories for chain_constant_memory
- TensorFlow util for building memory usage timeline from LOG_MEMORY messages☆65Updated 6 years ago
- Use TensorFlow efficiently☆95Updated 3 years ago
- Prunable nn layers for pytorch.☆48Updated 6 years ago
- PyProf2: PyTorch Profiling tool☆83Updated 4 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
- Just-in-time Dynamic Batching with MXNet Gluon.☆52Updated 4 years ago
- Odds and Ends and Things I've implemented.☆78Updated 5 years ago
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆39Updated 6 years ago
- Tensorflow implementation of SGD with Coupled Adaptive Batch Size (CABS)☆43Updated 7 years ago
- Training wide residual networks for deployment using a single bit for each weight☆36Updated 4 years ago
- Implementation of Neural Arithmetic Logic Units (https://arxiv.org/pdf/1808.00508.pdf)☆31Updated 6 years ago
- Python way to Read/Write TFRecords☆64Updated 6 years ago
- Simple example of implementing a new Tensorflow operation and its gradient in C++.☆56Updated 5 years ago
- Implementation of "Learning with Random Learning Rates" in PyTorch.☆102Updated 5 years ago
- OpenCL Inference Engine for pytorch☆51Updated 6 years ago
- PyTorch development for onnx☆21Updated 6 years ago
- ☆53Updated 6 years ago
- A more memory efficient Torch implementation of "Densely Connected Convolutional Networks".☆30Updated 7 years ago
- ☆58Updated 6 years ago
- DLPack for Tensorflow☆36Updated 4 years ago
- Direct C++ Interface to PyTorch☆80Updated 6 years ago
- Lasagne code for weight normalization☆87Updated 8 years ago
- ☆36Updated 5 years ago
- Distributed Learning by Pair-Wise Averaging☆53Updated 7 years ago
- Simple, efficient and flexible vision toolbox for mxnet framework.☆31Updated 6 years ago
- ONNX Integration Builds☆20Updated 6 years ago
- Weight initialization schemes for PyTorch nn.Modules☆70Updated 7 years ago
- ☆47Updated 4 years ago
- Various experiments on the [Fashion-MNIST](https://github.com/zalandoresearch/fashion-mnist) dataset from Zalando☆31Updated 7 years ago