sniklaus / pytorch-extensionLinks
an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors
☆118Updated 2 months ago
Alternatives and similar repositories for pytorch-extension
Users that are interested in pytorch-extension are comparing it to the libraries listed below
Sorting:
- CuPy fused PyTorch neural networks ops☆273Updated 7 years ago
- Photographic Image Synthesis with Cascaded Refinement Networks - Pytorch Implementation☆64Updated 7 years ago
- A PyTorch library for differentiable submodular minimization.☆58Updated 7 years ago
- Odds and Ends and Things I've implemented.☆78Updated 6 years ago
- PyTorch implementation of Wide Residual Networks with 1-bit weights by McDonnell (ICLR 2018)☆126Updated 6 years ago
- The repo of Feedback Networks, CVPR17☆90Updated 6 years ago
- PyTorch implementation of CVPR'18 - Perturbative Neural Networks http://xujuefei.com/pnn.html☆138Updated 6 years ago
- Binary Stochastic Neurons in PyTorch☆57Updated 7 years ago
- Repository has been moved: https://github.com/adobe/antialiased-cnns☆165Updated 3 years ago
- An implementation of shampoo☆77Updated 7 years ago
- [ECCV 2018] Sparsely Aggreagated Convolutional Networks https://arxiv.org/abs/1801.05895☆124Updated 6 years ago
- Code for "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis" https://arxiv.org/abs/1905.05934☆113Updated 5 years ago
- Code used to generate the results appearing in "Train longer, generalize better: closing the generalization gap in large batch training o…☆149Updated 8 years ago
- How and why you want to make your pytorch CUDA/CPP extension with a Makefile☆172Updated 6 years ago
- An intriguing failing of convolutional neural networks and the CoordConv solution in PyTorch☆72Updated 7 years ago
- Image-to-image translation with flow-based generative model☆64Updated 6 years ago
- Implements pytorch code for the Accelerated SGD algorithm.☆215Updated 7 years ago
- Pytorch code for our ICLR 2017 paper "Layered-Recursive GAN for image generation"☆151Updated 4 years ago
- Training wide residual networks for deployment using a single bit for each weight - Official Code Repository for ICLR 2018 Published Pape…☆37Updated 5 years ago
- All convolution operators replaced with coordinate convolutions☆36Updated 6 years ago
- Make python virtual environment setup on old servers less painful☆10Updated 7 years ago
- One Network to Solve Them All☆84Updated 6 years ago
- Implementation of our ECCV '18 paper " Deep Expander Networks: Efficient Deep Networks from Graph Theory".☆42Updated 5 years ago
- Code for Decorrelated Batch Normalization☆82Updated 7 years ago
- 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
- Learning Sparse High Dimensional Filters with Neural Networks☆69Updated 8 years ago
- A Pytorch implementation of "LegoNet: Efficient Convolutional Neural Networks with Lego Filters" (ICML 2019).☆140Updated 4 years ago
- Pytorch Implementation of OpenAI's GLOW☆94Updated 4 years ago
- Examples of C extensions for PyTorch☆256Updated 2 years ago
- ☆28Updated 5 years ago