Example repository for custom C++/CUDA operators for TorchScript
☆114Aug 28, 2022Updated 3 years ago
Alternatives and similar repositories for extension-script
Users that are interested in extension-script are comparing it to the libraries listed below
Sorting:
- C++ extensions in PyTorch☆1,185Jan 13, 2026Updated 2 months ago
- MSRSegNet: Multi-Scale Residual Network for Semantic Segmentation☆10Aug 9, 2018Updated 7 years ago
- Demonstration of using Caffe2 inside an Android application.☆68Dec 23, 2018Updated 7 years ago
- Implementation of Adversarial Variational Optimization in PyTorch☆43Aug 7, 2018Updated 7 years ago
- Parameterized fit and prediction harnesses for pytorch☆39Oct 23, 2020Updated 5 years ago
- Implementations of ideas from recent papers☆391Dec 22, 2020Updated 5 years ago
- A very naive and simple benchmark between dlib and pytorch in terms of space and time☆19May 13, 2020Updated 5 years ago
- CuPy fused PyTorch neural networks ops☆273Feb 15, 2018Updated 8 years ago
- high performance image loading and augmenting routines mimicking PIL.Image interface☆319Aug 16, 2021Updated 4 years ago
- Code for https://arxiv.org/abs/1810.04622☆140Aug 28, 2019Updated 6 years ago
- Fast radial symmetry transform implemented in pure python.☆13Apr 27, 2019Updated 6 years ago
- ☆13Aug 9, 2018Updated 7 years ago
- PyTorch elastic training☆729Jun 15, 2022Updated 3 years ago
- C++ PyTorch Examples☆10Aug 18, 2019Updated 6 years ago
- A faster python zipfile.☆31Feb 22, 2018Updated 8 years ago
- PyTorch layer-by-layer model profiler☆606May 23, 2021Updated 4 years ago
- Tutorial for building a custom CUDA function for Pytorch☆523Jan 25, 2019Updated 7 years ago
- Detectorch - detectron for PyTorch☆559Oct 30, 2018Updated 7 years ago
- TorchCV: a PyTorch vision library mimics ChainerCV☆396Dec 25, 2018Updated 7 years ago
- logboard: Monitor and Compare Logs on Browser/Terminal.☆21Sep 19, 2019Updated 6 years ago
- PyTorch implementation of PNASNet-5 on ImageNet☆320Aug 4, 2022Updated 3 years ago
- Slides from various talks I gave☆18Oct 25, 2018Updated 7 years ago
- A Pytorch implementation of "LegoNet: Efficient Convolutional Neural Networks with Lego Filters" (ICML 2019).☆140Oct 30, 2020Updated 5 years ago
- Pytorch implementation of MaxPoolingLoss.☆177Jun 9, 2018Updated 7 years ago
- [This project has moved to ChainerCV] Chainer Implementation of Light Head RCNN☆18May 8, 2020Updated 5 years ago
- Implements pytorch code for the Accelerated SGD algorithm.☆216Mar 10, 2018Updated 8 years ago
- Echo aware source separation☆13May 29, 2018Updated 7 years ago
- SplitNet implemented based on ResNet-50 trained on ImageNet-22K☆16Jun 18, 2018Updated 7 years ago
- How and why you want to make your pytorch CUDA/CPP extension with a Makefile☆172Jul 3, 2019Updated 6 years ago
- Code for ICML 2018 paper on "Fast and Scalable Bayesian Deep Learning by Weight-Perturbation in Adam" by Khan, Nielsen, Tangkaratt, Lin, …☆112Dec 19, 2018Updated 7 years ago
- An implementation of the Prism layer (https://arxiv.org/abs/2011.04823)☆12Nov 13, 2020Updated 5 years ago
- PyTorch implementation of Wide Residual Networks with 1-bit weights by McDonnell (ICLR 2018)☆126Sep 6, 2018Updated 7 years ago
- Implementation of the HMAX model of vision in PyTorch☆61May 26, 2023Updated 2 years ago
- PyTorch C++ Extension Example☆15Mar 4, 2018Updated 8 years ago
- Submission to https://pytorch.org/hub/☆1,432Apr 15, 2024Updated last year
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆683Feb 21, 2020Updated 6 years ago
- TVM integration into PyTorch☆456Jan 15, 2020Updated 6 years ago
- PyTorch 1.0 inference in C++ on Windows10 platforms☆89Jan 17, 2019Updated 7 years ago
- PyTorch Java demo☆35Nov 16, 2019Updated 6 years ago