sniklaus / pytorch-extension
an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors
☆117Updated 6 months ago
Related projects: ⓘ
- CuPy fused PyTorch neural networks ops☆274Updated 6 years ago
- How and why you want to make your pytorch CUDA/CPP extension with a Makefile☆171Updated 5 years ago
- Repository has been moved: https://github.com/adobe/antialiased-cnns☆164Updated 3 years ago
- PyTorch implementation of Wide Residual Networks with 1-bit weights by McDonnell (ICLR 2018)☆124Updated 6 years ago
- Odds and Ends and Things I've implemented.☆78Updated 5 years ago
- A PyTorch library for differentiable submodular minimization.☆57Updated 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
- Photographic Image Synthesis with Cascaded Refinement Networks - Pytorch Implementation☆65Updated 6 years ago
- Binary Stochastic Neurons in PyTorch☆56Updated 6 years ago
- ☆48Updated this week
- Structured Bayesian Pruning, NIPS 2017☆73Updated 4 years ago
- Pytorch Custom CUDA kernel for searchsorted☆136Updated 10 months ago
- ☆22Updated 5 years ago
- Minimal PyTorch implementation of Generative Latent Optimization from the paper "Optimizing the Latent Space of Generative Networks"☆120Updated 6 years ago
- Pytorch Implementation of OpenAI's GLOW☆92Updated 3 years ago
- Code for "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis" https://arxiv.org/abs/1905.05934☆111Updated 4 years ago
- Learning Sparse High Dimensional Filters with Neural Networks☆69Updated 7 years ago
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆39Updated 6 years ago
- Glow: Generative Flow with Invertible 1×1 Convolutions☆77Updated 5 years ago
- An implementation of shampoo☆73Updated 6 years ago
- Filter Response Normalization tested on better ImageNet baselines.☆35Updated 4 years ago
- The repo of Feedback Networks, CVPR17☆89Updated 5 years ago
- Pytorch code for our ICLR 2017 paper "Layered-Recursive GAN for image generation"☆151Updated 3 years ago
- ☆38Updated 7 years ago
- PyTorch implementation of CVPR'18 - Perturbative Neural Networks http://xujuefei.com/pnn.html☆138Updated 5 years ago
- [ECCV 2018] Sparsely Aggreagated Convolutional Networks https://arxiv.org/abs/1801.05895☆126Updated 5 years ago
- tunz's CUDA pytorch operator (MaskedSoftmax)☆74Updated 5 years ago
- Second-order optimiser for deep networks☆76Updated 5 years ago
- Code for Decorrelated Batch Normalization☆80Updated 6 years ago
- Examples of C extensions for PyTorch☆258Updated last year