Examples of C extensions for PyTorch
☆255Feb 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:
- an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors☆120May 26, 2025Updated last year
- C++ extensions in PyTorch☆1,203Oct 7, 2026Updated 3 weeks ago
- PyTorch C++ Extension Example☆15Mar 4, 2018Updated 8 years ago
- Tutorial for building a custom CUDA function for Pytorch☆526Jan 25, 2019Updated 7 years ago
- CuPy fused PyTorch neural networks ops☆273Feb 15, 2018Updated 8 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- ATen: A TENsor library for C++11☆722Nov 20, 2019Updated 6 years ago
- random pytorch hacks☆26Jul 26, 2017Updated 9 years ago
- PyTorch and Tensorflow functional model definitions☆584Feb 1, 2018Updated 8 years ago
- PyTorch implementation of Deformable Convolution☆409Feb 17, 2019Updated 7 years ago
- Autograd automatically differentiates native Torch code☆555Oct 21, 2021Updated 5 years ago
- ☆276Mar 12, 2017Updated 9 years ago
- Display images directly in iTerm2☆28May 20, 2016Updated 10 years ago
- OptNet - Reducing memory usage in torch neural nets☆282Apr 19, 2017Updated 9 years ago
- This library provides utilities for creating and manipulating RNNs to model sequential data.☆191Jul 28, 2017Updated 9 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- A language-agnostic interface to TensorBoard☆781Nov 9, 2017Updated 8 years ago
- An implementation of RetinaNet in PyTorch.☆305Sep 15, 2018Updated 8 years ago
- A CUDA backend for Torch7☆339Sep 11, 2017Updated 9 years ago
- A lightweight library for PyTorch training tools and utilities☆1,723Oct 29, 2026Updated last week
- Implementation of Adversarial Variational Optimization in PyTorch☆42Aug 7, 2018Updated 8 years ago
- Pytorch implementation of MaxPoolingLoss.☆177Jun 9, 2018Updated 8 years ago
- YOLOv2 in PyTorch☆1,561Sep 29, 2021Updated 5 years ago
- PyTorch implementation of Deformable Convolution☆904Jul 21, 2021Updated 5 years ago
- Training Very Deep Neural Networks Without Skip-Connections☆591Jun 9, 2018Updated 8 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- OpenCL Inference Engine for pytorch☆51Jan 13, 2018Updated 8 years ago
- A CV toolkit for my papers.☆2,045Sep 11, 2026Updated last month
- PyTorch implementation of "Soft Proposal Networks for Weakly Supervised Object Localization", ICCV 2017.☆207Jul 8, 2018Updated 8 years ago
- Implements an efficient softmax approximation as described in the paper "Efficient softmax approximation for GPUs" (http://arxiv.org/abs/…☆395Mar 22, 2019Updated 7 years ago
- A memory-efficient implementation of DenseNets☆1,536Jun 1, 2023Updated 3 years ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆8,001Jul 14, 2026Updated 3 months ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆9,000Updated this week
- Graph Computation for nn☆299Aug 1, 2017Updated 9 years ago
- Code for Switchable Normalization from "Differentiable Learning-to-Normalize via Switchable Normalization", https://arxiv.org/abs/1806.10…☆871Jun 11, 2020Updated 6 years ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.☆2,806Sep 5, 2019Updated 7 years ago
- CondenseNet: Light weighted CNN for mobile devices☆690Nov 11, 2019Updated 6 years ago
- Implementations of ideas from recent papers☆390Dec 22, 2020Updated 5 years ago
- Dual Path Networks (DPN) supporting pretrained weights converted from original MXNet implementation☆205Jun 11, 2019Updated 7 years ago
- DeepLab resnet v2 model in pytorch☆601Sep 5, 2023Updated 3 years ago
- A simple module for <Torch7> and the <nn> package☆18Apr 30, 2015Updated 11 years ago
- Write PyTorch code at the level of individual examples, then run it efficiently on minibatches.☆487Feb 12, 2022Updated 4 years ago