sjperkins / tfopgenLinks
Generate C++ and CUDA boilerplate for tensorflow custom operators
☆20Updated 7 years ago
Alternatives and similar repositories for tfopgen
Users that are interested in tfopgen are comparing it to the libraries listed below
Sorting:
- Training deep neural networks with low precision multiplications☆64Updated 10 years ago
- NNVM for ROCm Examples☆19Updated 8 years ago
- Benchmarks for CNTK and other toolkits.☆44Updated 10 years ago
- Distributed Learning by Pair-Wise Averaging☆52Updated 8 years ago
- Efficient layer normalization GPU kernel for Tensorflow☆110Updated 8 years ago
- The Operator Vectorization Library, or OVL, is a python productivity library for defining high performance custom operators for the Tenso…☆68Updated 9 years ago
- Tensorflow implementation of SGD with Coupled Adaptive Batch Size (CABS)☆44Updated 8 years ago
- Bugfixing fork of Python bindings for the NVIDIA GPU Management Library☆51Updated 8 years ago
- TensorFlow util for building memory usage timeline from LOG_MEMORY messages☆65Updated 8 years ago
- Direct C++ Interface to PyTorch☆81Updated 7 years ago
- Python wrappers for the NVIDIA cuDNN libraries☆142Updated 8 years ago
- A tutorial on 'Soft weight-sharing for Neural Network compression' published at ICLR2017☆147Updated 8 years ago
- Just-in-time Dynamic Batching with MXNet Gluon.☆52Updated 5 years ago
- Files to create the figures in the paper "Super-Convergence: Very Fast Training of Residual Networks Using Large Learning Rates"☆191Updated 8 years ago
- Reproduction of some of the results from 'Identity Mappings in Deep Residual Networks'☆72Updated 9 years ago
- Multi-core CPU implementation of deep learning for 2D and 3D sliding window convolutional networks (ConvNets).☆94Updated 9 years ago
- Tools to convert Caffe models to neon's serialization format☆39Updated 2 years ago
- PyTorch development for onnx☆21Updated 8 years ago
- ☆22Updated 9 years ago
- Source code for ``Neural Networks with Few Multiplications'' published at ICLR 2016☆80Updated 9 years ago
- Sublinear memory optimization for deep learning, reduce GPU memory cost to train deeper nets☆28Updated 9 years ago
- Binarized Neural Network TF training code + C matrix / eval library.☆101Updated 8 years ago
- MPI Parallel framework for training deep learning models built in Theano☆54Updated 8 years ago
- Stuff I uploaded to share online or to access from a different machine☆174Updated 3 years ago
- Torch FFI-bindings for NNPACK☆31Updated 8 years ago
- ☆37Updated 6 years ago
- neon implementation of SegNet☆13Updated 2 years ago
- ☆47Updated 5 years ago
- Python Binding to NVRTC☆79Updated last year
- 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