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
- A plug-in replacement for DataLoader to load Imagenet disk-sequentially in PyTorch.☆238Updated 3 years ago
- an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors☆118Updated 2 months ago
- PyTorch implementation of Deformable Convolution☆408Updated 6 years ago
- Examples of C extensions for PyTorch☆257Updated 2 years ago
- PyTorch wrapper for FFTs☆315Updated 6 years ago
- PyTorch implementation of Wide Residual Networks with 1-bit weights by McDonnell (ICLR 2018)☆124Updated 6 years ago
- Repository has been moved: https://github.com/adobe/antialiased-cnns☆165Updated 3 years ago
- pytorch model summary, statistic parameters number, memory usage, FLOPs and so on☆97Updated 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
- PyTorch and Tensorflow functional model definitions☆585Updated 7 years ago
- PyTorch implementation for Convolutional Networks with Adaptive Inference Graphs☆185Updated 6 years ago
- Multi-Scale Dense Networks for Resource Efficient Image Classification (ICLR 2018 Oral)☆461Updated 5 years ago
- ICLR 2018 reproducibility challenge - Multi-Scale Dense Convolutional Networks for Efficient Prediction☆135Updated 6 years ago
- Code for "The Reversible Residual Network: Backpropagation Without Storing Activations"☆357Updated 6 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
- CondenseNet: Light weighted CNN for mobile devices☆693Updated 5 years ago
- PyTorch implementation of PNASNet-5 on ImageNet☆317Updated 2 years ago
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆77Updated 6 years ago
- The repo of Feedback Networks, CVPR17☆89Updated 6 years ago
- Low-rank convolutional neural networks☆96Updated 8 years ago
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆38Updated 6 years ago
- How and why you want to make your pytorch CUDA/CPP extension with a Makefile☆172Updated 5 years ago
- ☆165Updated 6 years ago
- repo that holds code for improving on dropout using Stochastic Delta Rule☆142Updated 6 years ago
- Pytorch implementation of CoordConv introduced in 'An intriguing failing of convolutional neural networks and the CoordConv solution' pap…☆399Updated 6 years ago
- A Re-implementation of Fixed-update Initialization☆152Updated 5 years ago
- Oriented Response Networks, in CVPR 2017☆221Updated 2 years ago
- Sparse Blocks Networks☆433Updated 6 years ago
- TorchCV: a PyTorch vision library mimics ChainerCV☆394Updated 6 years ago