sjperkins / tfopgen
Generate C++ and CUDA boilerplate for tensorflow custom operators
☆20Updated 6 years ago
Alternatives and similar repositories for tfopgen:
Users that are interested in tfopgen are comparing it to the libraries listed below
- Distributed Learning by Pair-Wise Averaging☆52Updated 7 years ago
- ☆36Updated 6 years ago
- Tensorflow implementation of SGD with Coupled Adaptive Batch Size (CABS)☆43Updated 8 years ago
- A tutorial on 'Soft weight-sharing for Neural Network compression' published at ICLR2017☆145Updated 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
- Python bindings for pyNVML and psutil library over network☆51Updated last year
- Efficient layer normalization GPU kernel for Tensorflow☆111Updated 8 years ago
- Deep learning with a multiplication budget☆47Updated 6 years ago
- easy embeddable Torch7 networks☆35Updated 8 years ago
- MPI for Torch☆60Updated 7 years ago
- MPI Parallel framework for training deep learning models built in Theano☆54Updated 7 years ago
- train on AWS☆75Updated 6 years ago
- Sublinear memory optimization for deep learning, reduce GPU memory cost to train deeper nets☆28Updated 9 years ago
- Direct C++ Interface to PyTorch☆80Updated 7 years ago
- Implementation of "Learning with Random Learning Rates" in PyTorch.☆102Updated 5 years ago
- Benchmarks for CNTK and other toolkits.☆44Updated 9 years ago
- Torch FFI-bindings for NNPACK☆30Updated 7 years ago
- Files to create the figures in the paper "Super-Convergence: Very Fast Training of Residual Networks Using Large Learning Rates"☆191Updated 7 years ago
- ☆57Updated 9 years ago
- TensorFlow util for building memory usage timeline from LOG_MEMORY messages☆64Updated 7 years ago
- Training deep neural networks with low precision multiplications☆63Updated 9 years ago
- Small Python library to automatically set CUDA_VISIBLE_DEVICES to the least loaded device on multi-GPU systems.☆106Updated 2 years ago
- Reproduction of some of the results from 'Identity Mappings in Deep Residual Networks'☆72Updated 8 years ago
- Just-in-time Dynamic Batching with MXNet Gluon.☆52Updated 4 years ago
- Boda: A C++ Framework for Efficient Experiments in Computer Vision☆64Updated 5 years ago
- Code used to generate the results appearing in "Train longer, generalize better: closing the generalization gap in large batch training o…☆149Updated 7 years ago
- DeepArchitect: Automatically Designing and Training Deep Architectures☆145Updated 5 years ago
- NNVM for ROCm Examples☆19Updated 7 years ago
- Odds and Ends and Things I've implemented.☆78Updated 6 years ago
- DLPack for Tensorflow☆35Updated 5 years ago