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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- C++ extensions in PyTorch☆1,186Jan 13, 2026Updated 2 months 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
- CuPy fused PyTorch neural networks ops☆273Feb 15, 2018Updated 8 years ago
- ATen: A TENsor library for C++11☆717Nov 20, 2019Updated 6 years ago
- Tutorial for building a custom CUDA function for Pytorch☆523Jan 25, 2019Updated 7 years ago
- PyTorch and Tensorflow functional model definitions☆588Feb 1, 2018Updated 8 years ago
- This library provides utilities for creating and manipulating RNNs to model sequential data.☆191Jul 28, 2017Updated 8 years ago
- random pytorch hacks☆26Jul 26, 2017Updated 8 years ago
- Display images directly in iTerm2☆28May 20, 2016Updated 9 years ago
- PyTorch implementation of Deformable Convolution☆411Feb 17, 2019Updated 7 years ago
- ☆278Mar 12, 2017Updated 9 years ago
- OptNet - Reducing memory usage in torch neural nets☆282Apr 19, 2017Updated 8 years ago
- A language-agnostic interface to TensorBoard☆780Nov 9, 2017Updated 8 years ago
- Autograd automatically differentiates native Torch code☆558Oct 21, 2021Updated 4 years ago
- OpenCL Inference Engine for pytorch☆51Jan 13, 2018Updated 8 years ago
- A simple module for <Torch7> and the <nn> package☆18Apr 30, 2015Updated 10 years ago
- Detectorch - detectron for PyTorch☆559Oct 30, 2018Updated 7 years ago
- PyTorch implementation of "Soft Proposal Networks for Weakly Supervised Object Localization", ICCV 2017.☆210Jul 8, 2018Updated 7 years ago
- An implementation of RetinaNet in PyTorch.☆304Sep 15, 2018Updated 7 years ago
- load word embeddings to Torch.Tensor☆14May 12, 2016Updated 9 years ago
- A memory-efficient implementation of DenseNets☆1,541Jun 1, 2023Updated 2 years ago
- Training Very Deep Neural Networks Without Skip-Connections☆589Jun 9, 2018Updated 7 years ago
- A curated list of awesome Torch tutorials, projects and communities☆648Mar 22, 2018Updated 8 years ago
- Implements an efficient softmax approximation as described in the paper "Efficient softmax approximation for GPUs" (http://arxiv.org/abs/…☆396Mar 22, 2019Updated 7 years ago
- Write PyTorch code at the level of individual examples, then run it efficiently on minibatches.☆485Feb 12, 2022Updated 4 years ago
- Tools for PyTorch☆224Aug 10, 2022Updated 3 years ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,990Feb 5, 2026Updated last month
- Code for the Torch in Action book☆44Jan 24, 2017Updated 9 years ago
- Code for Switchable Normalization from "Differentiable Learning-to-Normalize via Switchable Normalization", https://arxiv.org/abs/1806.10…☆869Jun 11, 2020Updated 5 years ago
- Pytorch implementation of MaxPoolingLoss.☆177Jun 9, 2018Updated 7 years ago
- [adversarial] examples and training cost☆19Jun 29, 2016Updated 9 years ago
- YOLOv2 in PyTorch☆1,564Sep 29, 2021Updated 4 years ago
- A domain specific language to express machine learning workloads.☆1,764Apr 28, 2023Updated 2 years ago
- PyTorch implementation of the Quasi-Recurrent Neural Network - up to 16 times faster than NVIDIA's cuDNN LSTM☆1,264Feb 12, 2022Updated 4 years ago
- DeepLab resnet v2 model in pytorch☆603Sep 5, 2023Updated 2 years ago
- Graph Computation for nn☆302Aug 1, 2017Updated 8 years ago
- A CV toolkit for my papers.☆2,048Dec 21, 2024Updated last year
- A PyTorch Implementation for Densely Connected Convolutional Networks (DenseNets)☆487Feb 28, 2018Updated 8 years ago