szagoruyko / pyinn
CuPy fused PyTorch neural networks ops
☆273Updated 7 years ago
Alternatives and similar repositories for pyinn:
Users that are interested in pyinn are comparing it to the libraries listed below
- Examples of C extensions for PyTorch☆256Updated 2 years ago
- an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors☆118Updated 3 months ago
- A plug-in replacement for DataLoader to load Imagenet disk-sequentially in PyTorch.☆238Updated 3 years ago
- PyTorch implementation of Deformable Convolution☆408Updated 6 years ago
- PyTorch implementation of Wide Residual Networks with 1-bit weights by McDonnell (ICLR 2018)☆123Updated 6 years ago
- pytorch model summary, statistic parameters number, memory usage, FLOPs and so on☆98Updated 5 years ago
- Repository has been moved: https://github.com/adobe/antialiased-cnns☆165Updated 3 years ago
- PyTorch wrapper for FFTs☆316Updated 6 years ago
- ICLR 2018 reproducibility challenge - Multi-Scale Dense Convolutional Networks for Efficient Prediction☆135Updated 6 years ago
- PyTorch implementation of PNASNet-5 on ImageNet☆317Updated 2 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
- Code for "The Reversible Residual Network: Backpropagation Without Storing Activations"☆359Updated 6 years ago
- visualization of CNN in PyTorch☆155Updated last year
- [ECCV 2018] Sparsely Aggreagated Convolutional Networks https://arxiv.org/abs/1801.05895☆124Updated 6 years ago
- PyTorch implementation for Convolutional Networks with Adaptive Inference Graphs☆185Updated 6 years ago
- Spatially Adaptive Computation Time for Residual Networks☆248Updated 2 years ago
- The repo of Feedback Networks, CVPR17☆89Updated 6 years ago
- A tutorial on 'Soft weight-sharing for Neural Network compression' published at ICLR2017☆145Updated 8 years ago
- SE-Net Incorporates with ResNet and WideResnet on CIFAR-10/100 Dataset.☆47Updated 7 years ago
- Low-rank convolutional neural networks☆97Updated 9 years ago
- Multi-Scale Dense Networks for Resource Efficient Image Classification (ICLR 2018 Oral)☆460Updated 6 years ago
- ☆251Updated 8 years ago
- repo that holds code for improving on dropout using Stochastic Delta Rule☆142Updated 6 years ago
- PyTorch and Tensorflow functional model definitions☆585Updated 7 years ago
- Training Very Deep Neural Networks Without Skip-Connections☆586Updated 6 years ago
- Implementation of OctConv in Pytorch (https://arxiv.org/abs/1904.05049)☆214Updated 5 years ago
- Path-Level Network Transformation for Efficient Architecture Search, in ICML 2018.☆112Updated 6 years ago
- Implements pytorch code for the Accelerated SGD algorithm.☆215Updated 7 years ago
- ☆165Updated 6 years ago
- A memory efficient implementation of densenet.☆82Updated 5 years ago