sjperkins / tfopgenLinks
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
Sorting:
- NNVM for ROCm Examples☆19Updated 7 years ago
- PyTorch development for onnx☆21Updated 7 years ago
- Python Binding to NVRTC☆79Updated 7 months ago
- Distributed Learning by Pair-Wise Averaging☆52Updated 7 years ago
- ☆37Updated 6 years ago
- Boda: A C++ Framework for Efficient Experiments in Computer Vision☆64Updated 5 years ago
- Tools to convert Caffe models to neon's serialization format☆39Updated 2 years ago
- TensorFlow util for building memory usage timeline from LOG_MEMORY messages☆64Updated 7 years ago
- Torch FFI-bindings for NNPACK☆30Updated 8 years ago
- DelugeNets: Deep Networks with Efficient and Flexible Cross-layer Information Inflows☆26Updated 8 years ago
- Sublinear memory optimization for deep learning, reduce GPU memory cost to train deeper nets☆28Updated 9 years ago
- easy embeddable Torch7 networks☆35Updated 8 years ago
- Convolution op for Theano based on CuFFT using scikits.cuda☆52Updated 10 years ago
- Tensorflow implementation of SGD with Coupled Adaptive Batch Size (CABS)☆43Updated 8 years ago
- Efficient layer normalization GPU kernel for Tensorflow☆111Updated 8 years ago
- Multi-core CPU implementation of deep learning for 2D and 3D sliding window convolutional networks (ConvNets).☆94Updated 8 years ago
- Reproduction of some of the results from 'Identity Mappings in Deep Residual Networks'☆72Updated 8 years ago
- Training deep neural networks with low precision multiplications☆63Updated 9 years ago
- A rudimentary wrapper around the fast Maxwell kernels for GEMM and convolution operations provided by nervanagpu☆34Updated 10 years ago
- ☆23Updated 8 years ago
- Just-in-time Dynamic Batching with MXNet Gluon.☆52Updated 5 years ago
- DNI(Decoupled Neural Interfaces using Synthetic Gradients) implementation with Torch☆29Updated 8 years ago
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆38Updated 6 years ago
- DLPack for Tensorflow☆35Updated 5 years ago
- Demos interesting image-in, image-out networks running on both NVIDIA and AMD GPUs, with NNVM☆49Updated 7 years ago
- Functional ANOVA☆27Updated 10 years ago
- MXNet Model Serving☆25Updated 7 years ago
- Convolution implementation with on the fly Toeplitz matrix generation.☆11Updated 8 years ago
- Simple MXNet sequence-to-sequence model (neural machine translation)☆24Updated 7 years ago
- Python bindings for pyNVML and psutil library over network☆51Updated last year