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,184Jan 13, 2026Updated last month
- 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
- C++ PyTorch Examples☆10Aug 18, 2019Updated 6 years ago
- MSRSegNet: Multi-Scale Residual Network for Semantic Segmentation☆10Aug 9, 2018Updated 7 years ago
- Implementations of ideas from recent papers☆392Dec 22, 2020Updated 5 years ago
- Code for https://arxiv.org/abs/1810.04622☆140Aug 28, 2019Updated 6 years ago
- high performance image loading and augmenting routines mimicking PIL.Image interface☆318Aug 16, 2021Updated 4 years ago
- ☆13Aug 9, 2018Updated 7 years ago
- Parameterized fit and prediction harnesses for pytorch☆39Oct 23, 2020Updated 5 years ago
- PyTorch layer-by-layer model profiler☆607May 23, 2021Updated 4 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
- Implementation of the HMAX model of vision in PyTorch☆61May 26, 2023Updated 2 years ago
- [This project has moved to ChainerCV] Chainer Implementation of Light Head RCNN☆18May 8, 2020Updated 5 years ago
- SplitNet implemented based on ResNet-50 trained on ImageNet-22K☆16Jun 18, 2018Updated 7 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
- A Pytorch implementation of "LegoNet: Efficient Convolutional Neural Networks with Lego Filters" (ICML 2019).☆140Oct 30, 2020Updated 5 years ago
- PyTorch elastic training☆728Jun 15, 2022Updated 3 years ago
- logboard: Monitor and Compare Logs on Browser/Terminal.☆21Sep 19, 2019Updated 6 years ago
- TorchCV: a PyTorch vision library mimics ChainerCV☆396Dec 25, 2018Updated 7 years ago
- Detectorch - detectron for PyTorch☆558Oct 30, 2018Updated 7 years ago
- Tutorial for building a custom CUDA function for Pytorch☆522Jan 25, 2019Updated 7 years ago
- Implements pytorch code for the Accelerated SGD algorithm.☆216Mar 10, 2018Updated 7 years ago
- Slides from various talks I gave☆18Oct 25, 2018Updated 7 years ago
- Pytorch implementation of MaxPoolingLoss.☆177Jun 9, 2018Updated 7 years ago
- PyTorch implementation of Wide Residual Networks with 1-bit weights by McDonnell (ICLR 2018)☆126Sep 6, 2018Updated 7 years ago
- PyTorch implementation of PNASNet-5 on ImageNet☆320Aug 4, 2022Updated 3 years ago
- A faster python zipfile.☆31Feb 22, 2018Updated 8 years ago
- PyTorch C++ Extension Example☆15Mar 4, 2018Updated 7 years ago
- A parallel evaluation data set of SAP software documentation with document structure annotation☆14Jul 30, 2025Updated 6 months ago
- How and why you want to make your pytorch CUDA/CPP extension with a Makefile☆172Jul 3, 2019Updated 6 years ago
- model uncertainty using mc dropout☆20Mar 1, 2019Updated 6 years ago
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆683Feb 21, 2020Updated 6 years ago
- Translate - a PyTorch Language Library☆834Apr 27, 2023Updated 2 years ago
- Submission to https://pytorch.org/hub/☆1,432Apr 15, 2024Updated last year
- Tools for PyTorch☆223Aug 10, 2022Updated 3 years ago
- SCOPS: Self-Supervised Co-Part Segmentation (CVPR'19)☆218Dec 8, 2022Updated 3 years ago
- Fast radial symmetry transform implemented in pure python.☆13Apr 27, 2019Updated 6 years ago