Examples of C extensions for PyTorch
☆256Feb 12, 2023Updated 3 years ago
Alternatives and similar repositories for extension-ffi
Users that are interested in extension-ffi are comparing it to the libraries listed below
Sorting:
- C++ extensions in PyTorch☆1,184Jan 13, 2026Updated last month
- CuPy fused PyTorch neural networks ops☆273Feb 15, 2018Updated 8 years ago
- an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors☆119May 26, 2025Updated 9 months ago
- PyTorch C++ Extension Example☆15Mar 4, 2018Updated 8 years ago
- ATen: A TENsor library for C++11☆717Nov 20, 2019Updated 6 years ago
- PyTorch and Tensorflow functional model definitions☆588Feb 1, 2018Updated 8 years ago
- Tutorial for building a custom CUDA function for Pytorch☆522Jan 25, 2019Updated 7 years ago
- PyTorch implementation of Deformable Convolution☆411Feb 17, 2019Updated 7 years ago
- OpenCL Inference Engine for pytorch☆51Jan 13, 2018Updated 8 years ago
- ☆278Mar 12, 2017Updated 8 years ago
- Write PyTorch code at the level of individual examples, then run it efficiently on minibatches.☆485Feb 12, 2022Updated 4 years ago
- OptNet - Reducing memory usage in torch neural nets☆282Apr 19, 2017Updated 8 years ago
- random pytorch hacks☆26Jul 26, 2017Updated 8 years ago
- This library provides utilities for creating and manipulating RNNs to model sequential data.☆191Jul 28, 2017Updated 8 years ago
- PyTorch implementation of "Soft Proposal Networks for Weakly Supervised Object Localization", ICCV 2017.☆210Jul 8, 2018Updated 7 years ago
- Display images directly in iTerm2☆28May 20, 2016Updated 9 years ago
- An implementation of RetinaNet in PyTorch.☆303Sep 15, 2018Updated 7 years ago
- A language-agnostic interface to TensorBoard☆781Nov 9, 2017Updated 8 years ago
- Autograd automatically differentiates native Torch code☆559Oct 21, 2021Updated 4 years ago
- Tools for PyTorch☆223Aug 10, 2022Updated 3 years ago
- Detectorch - detectron for PyTorch☆559Oct 30, 2018Updated 7 years ago
- A memory-efficient implementation of DenseNets☆1,539Jun 1, 2023Updated 2 years ago
- Training Very Deep Neural Networks Without Skip-Connections☆589Jun 9, 2018Updated 7 years ago
- A utility library around PyTorch☆244Dec 26, 2022Updated 3 years ago
- A domain specific language to express machine learning workloads.☆1,765Apr 28, 2023Updated 2 years ago
- PyTorch implementation of the CortexNet predictive model☆368Nov 28, 2018Updated 7 years ago
- PyTorch implementation of the Quasi-Recurrent Neural Network - up to 16 times faster than NVIDIA's cuDNN LSTM☆1,265Feb 12, 2022Updated 4 years ago
- DeepLab resnet v2 model in pytorch☆603Sep 5, 2023Updated 2 years ago
- Implements an efficient softmax approximation as described in the paper "Efficient softmax approximation for GPUs" (http://arxiv.org/abs/…☆396Mar 22, 2019Updated 6 years ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,989Feb 5, 2026Updated 3 weeks ago
- Pytorch Implementation for Deeplab☆15May 1, 2018Updated 7 years ago
- A simple module for <Torch7> and the <nn> package☆18Apr 30, 2015Updated 10 years ago
- A CV toolkit for my papers.☆2,049Dec 21, 2024Updated last year
- YOLOv2 in PyTorch☆1,563Sep 29, 2021Updated 4 years ago
- Dual Path Networks (DPN) supporting pretrained weights converted from original MXNet implementation☆205Jun 11, 2019Updated 6 years ago
- Code for Switchable Normalization from "Differentiable Learning-to-Normalize via Switchable Normalization", https://arxiv.org/abs/1806.10…☆870Jun 11, 2020Updated 5 years ago
- Pytorch implementation of MaxPoolingLoss.☆177Jun 9, 2018Updated 7 years ago
- Code for "On the Effects of Batch and Weight Normalization in Generative Adversarial Networks"☆181Jan 15, 2018Updated 8 years ago
- DNN Inference with CPU, C++, ONNX support: Instant☆56Oct 17, 2018Updated 7 years ago