CuPy fused PyTorch neural networks ops
☆273Feb 15, 2018Updated 8 years ago
Alternatives and similar repositories for pyinn
Users that are interested in pyinn are comparing it to the libraries listed below
Sorting:
- an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors☆119May 26, 2025Updated 9 months ago
- auto-tuning momentum SGD optimizer☆288Mar 24, 2019Updated 6 years ago
- PyTorch implementation of Deformable Convolution☆411Feb 17, 2019Updated 7 years ago
- Tutorial for building a custom CUDA function for Pytorch☆522Jan 25, 2019Updated 7 years ago
- Fast Scattering Transform with CuPy/PyTorch☆295Feb 22, 2020Updated 6 years ago
- PyTorch bindings for openai-gemm☆20Feb 6, 2017Updated 9 years ago
- Training Very Deep Neural Networks Without Skip-Connections☆589Jun 9, 2018Updated 7 years ago
- Deeper DCGAN with AE stabilization☆38Mar 20, 2024Updated last year
- PyTorch implementation of Deformable Convolution☆911Jul 21, 2021Updated 4 years ago
- Pytorch for Segmentation☆414Jul 19, 2019Updated 6 years ago
- Pytorch implementation of "Forward Thinking: Building and Training Neural Networks One Layer at a Time"☆65Jun 14, 2017Updated 8 years ago
- high performance image loading and augmenting routines mimicking PIL.Image interface☆318Aug 16, 2021Updated 4 years ago
- In-Place Activated BatchNorm for Memory-Optimized Training of DNNs☆1,335Jul 8, 2025Updated 7 months ago
- PyTorch implementation of Wide Residual Networks with 1-bit weights by McDonnell (ICLR 2018)☆126Sep 6, 2018Updated 7 years ago
- Examples of C extensions for PyTorch☆256Feb 12, 2023Updated 3 years ago
- Segmentation-Aware Convolutional Networks Using Local Attention Masks☆146Nov 13, 2017Updated 8 years ago
- C++ extensions in PyTorch☆1,184Jan 13, 2026Updated last month
- OpenCL Inference Engine for pytorch☆51Jan 13, 2018Updated 8 years ago
- Dual Path Networks (DPN) supporting pretrained weights converted from original MXNet implementation☆205Jun 11, 2019Updated 6 years ago
- Generalized Compressed Network Search with PyTorch☆26Nov 1, 2017Updated 8 years ago
- A language-agnostic interface to TensorBoard☆781Nov 9, 2017Updated 8 years ago
- Dilated Residual Networks☆1,104Mar 9, 2020Updated 5 years ago
- PyTorch implementation of PNASNet-5 on ImageNet☆320Aug 4, 2022Updated 3 years ago
- ATen: A TENsor library for C++11☆717Nov 20, 2019Updated 6 years ago
- A lightweight library for PyTorch training tools and utilities☆1,720Feb 10, 2026Updated 3 weeks ago
- CondenseNet: Light weighted CNN for mobile devices☆691Nov 11, 2019Updated 6 years ago
- Active Convolution☆60Apr 9, 2017Updated 8 years ago
- PyTorch and Tensorflow functional model definitions☆588Feb 1, 2018Updated 8 years ago
- A plug-in replacement for DataLoader to load Imagenet disk-sequentially in PyTorch.☆239Aug 18, 2021Updated 4 years ago
- PyTorch C++ Extension Example☆15Mar 4, 2018Updated 8 years ago
- 2.86% and 15.85% on CIFAR-10 and CIFAR-100☆297Oct 9, 2018Updated 7 years ago
- A Caffe implementation of http://arxiv.org/abs/1512.07928☆39Jun 25, 2016Updated 9 years ago
- A PyTorch implementation of the paper "Semantic Image Synthesis via Adversarial Learning" in ICCV 2017☆142Aug 24, 2017Updated 8 years ago
- Decoupled Neural Interfaces using Synthetic Gradients for PyTorch☆239Jan 12, 2019Updated 7 years ago
- An implementation of RetinaNet in PyTorch.☆303Sep 15, 2018Updated 7 years ago
- An official collection of code in different frameworks that reproduces experiments in "Group Normalization"☆118Feb 23, 2021Updated 5 years ago
- Detectorch - detectron for PyTorch☆559Oct 30, 2018Updated 7 years ago
- A utility library around PyTorch☆244Dec 26, 2022Updated 3 years ago
- YOLOv2 in PyTorch☆1,563Sep 29, 2021Updated 4 years ago