pytorch / extension-ffi
Examples of C extensions for PyTorch
☆257Updated last year
Alternatives and similar repositories for extension-ffi:
Users that are interested in extension-ffi are comparing it to the libraries listed below
- CuPy fused PyTorch neural networks ops☆274Updated 6 years ago
- Sublinear memory optimization for deep learning, reduce GPU memory cost to train deeper nets☆308Updated 7 years ago
- auto-tuning momentum SGD optimizer☆287Updated 5 years ago
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆38Updated 6 years ago
- A plug-in replacement for DataLoader to load Imagenet disk-sequentially in PyTorch.☆238Updated 3 years ago
- Implements pytorch code for the Accelerated SGD algorithm.☆215Updated 6 years ago
- PyTorch and Tensorflow functional model definitions☆586Updated 6 years ago
- an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors☆118Updated 3 weeks ago
- PyTorch wrapper for FFTs☆315Updated 6 years ago
- 2.86% and 15.85% on CIFAR-10 and CIFAR-100☆296Updated 6 years ago
- Example code for Weight Normalization, from "Weight Normalization: A Simple Reparameterization to Accelerate Training of Deep Neural Netw…☆364Updated 6 years ago
- PyTorch implementation of PNASNet-5 on ImageNet☆317Updated 2 years ago
- Low-rank convolutional neural networks☆96Updated 8 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
- A tutorial on 'Soft weight-sharing for Neural Network compression' published at ICLR2017☆145Updated 7 years ago
- Spatially Adaptive Computation Time for Residual Networks☆247Updated last year
- This is an example of how to train a MNIST network in Python and run it in c++ with pytorch 1.0☆96Updated 6 years ago
- Path-Level Network Transformation for Efficient Architecture Search, in ICML 2018.☆112Updated 6 years ago
- Simple example of implementing a new Tensorflow operation and its gradient in C++.☆56Updated 5 years ago
- Sparse Blocks Networks☆434Updated 6 years ago
- Pytorch C++ Library☆369Updated 6 years ago
- ☆138Updated 8 years ago
- ICLR 2018 reproducibility challenge - Multi-Scale Dense Convolutional Networks for Efficient Prediction☆136Updated 6 years ago
- Caffe implementation for dynamic network surgery.☆186Updated 7 years ago
- Caffe for Sparse Convolutional Neural Network☆238Updated 2 years ago
- OptNet - Reducing memory usage in torch neural nets☆282Updated 7 years ago
- (New version is out: https://github.com/hpi-xnor/BMXNet-v2) BMXNet: An Open-Source Binary Neural Network Implementation Based on MXNet☆351Updated 5 years ago
- [ECCV 2018] Sparsely Aggreagated Convolutional Networks https://arxiv.org/abs/1801.05895☆124Updated 6 years ago
- visualization of CNN in PyTorch☆155Updated last year
- ☆251Updated 8 years ago