yaroslavvb / chain_constant_memory
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
- TensorFlow util for building memory usage timeline from LOG_MEMORY messages☆64Updated 7 years ago
- Python way to Read/Write TFRecords☆64Updated 6 years ago
- Use TensorFlow efficiently☆95Updated 3 years ago
- Distributed Learning by Pair-Wise Averaging☆53Updated 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
- PyProf2: PyTorch Profiling tool☆82Updated 4 years ago
- Simple example of implementing a new Tensorflow operation and its gradient in C++.☆56Updated 5 years ago
- Just-in-time Dynamic Batching with MXNet Gluon.☆52Updated 4 years ago
- Implementation of Neural Arithmetic Logic Units (https://arxiv.org/pdf/1808.00508.pdf)☆31Updated 6 years ago
- Compare outputs between layers written in Tensorflow and layers written in Pytorch☆72Updated 6 years ago
- Tensorflow implementation of SGD with Coupled Adaptive Batch Size (CABS)☆43Updated 8 years ago
- an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors☆118Updated 2 months ago
- Odds and Ends and Things I've implemented.☆78Updated 6 years ago
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆38Updated 6 years ago
- ☆57Updated 6 years ago
- Implementation of "Learning with Random Learning Rates" in PyTorch.☆102Updated 5 years ago
- Second-order optimiser for deep networks☆76Updated 6 years ago
- Prunable nn layers for pytorch.☆48Updated 6 years ago
- DLPack for Tensorflow☆36Updated 4 years ago
- "Recurrent Models of Visual Attention" in TensorFlow☆41Updated 7 years ago
- Simple implementation of the LSUV initialization in PyTorch☆58Updated last year
- Pytorch implementation of bytenet from "Neural Machine Translation in Linear Time" paper☆46Updated 7 years ago
- The latest version of Net-Trim which solves the actual constrained problem☆24Updated 4 years ago
- Implements an infinite sum of poisson-weighted convolutions☆26Updated 6 years ago
- Implementation of Adversarial Variational Optimization in PyTorch☆43Updated 6 years ago
- A new kind of pooling layer for faster and sharper convergence☆76Updated 7 years ago
- ONNX Integration Builds☆20Updated 6 years ago
- "Layer-wise Adaptive Rate Scaling" in PyTorch☆86Updated 4 years ago
- Python Binding to NVRTC☆79Updated 5 months ago
- Direct C++ Interface to PyTorch☆80Updated 6 years ago