an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors
☆119May 26, 2025Updated 10 months ago
Alternatives and similar repositories for pytorch-extension
Users that are interested in pytorch-extension are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- CuPy fused PyTorch neural networks ops☆273Feb 15, 2018Updated 8 years ago
- PyTorch C++ Extension Example☆15Mar 4, 2018Updated 8 years ago
- ScatWave is a Torch implementation of scattering using CUDA☆21Jan 18, 2018Updated 8 years ago
- ☆45Apr 25, 2017Updated 8 years ago
- Generalized Compressed Network Search with PyTorch☆26Nov 1, 2017Updated 8 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Convolutional network training using Torch☆18Sep 18, 2016Updated 9 years ago
- VLFeat (partial) FFI wrapper for Torch7☆12Mar 23, 2016Updated 10 years ago
- Tutorial for building a custom CUDA function for Pytorch☆523Jan 25, 2019Updated 7 years ago
- DNI(Decoupled Neural Interfaces using Synthetic Gradients) implementation with Torch☆30Aug 30, 2016Updated 9 years ago
- A Thrift codec for Torch☆30Oct 21, 2021Updated 4 years ago
- Deeper DCGAN with AE stabilization☆38Mar 20, 2024Updated 2 years ago
- Pytorch implementation of "Forward Thinking: Building and Training Neural Networks One Layer at a Time"☆65Jun 14, 2017Updated 8 years ago
- Examples of C extensions for PyTorch☆256Feb 12, 2023Updated 3 years ago
- Experiments with multiple classifiers to classify facial expressions☆14Feb 26, 2016Updated 10 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Online Lua REPL with Torch support.☆17Aug 28, 2015Updated 10 years ago
- random pytorch hacks☆26Jul 26, 2017Updated 8 years ago
- LMDB for Torch☆27Sep 4, 2017Updated 8 years ago
- lua apply function for cutorch☆17Jan 5, 2017Updated 9 years ago
- Generate captions for an image using convolutional and recurrent networks☆12Feb 25, 2016Updated 10 years ago
- ☆82Oct 19, 2017Updated 8 years ago
- Nervana Neon kernels in Torch☆19Nov 5, 2015Updated 10 years ago
- Torch FFI-bindings for NNPACK☆31May 26, 2017Updated 8 years ago
- Tools for PyTorch☆224Aug 10, 2022Updated 3 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Data structures, algorithms, and ML/NLP tools in Lua.☆26Aug 31, 2017Updated 8 years ago
- Training Very Deep Neural Networks Without Skip-Connections☆589Jun 9, 2018Updated 7 years ago
- an implementation of Video Frame Interpolation via Adaptive Separable Convolution using PyTorch☆1,024May 26, 2025Updated 10 months ago
- Unsupervised Learning of Multi-Frame Optical Flow with Occlusions☆112Nov 7, 2019Updated 6 years ago
- ☆222Feb 15, 2020Updated 6 years ago
- A neural assembly compiler for pyTorch based on adaptive-neural-compilation☆27Mar 27, 2018Updated 8 years ago
- PyTorch wrapper for FFTs☆317Oct 28, 2018Updated 7 years ago
- autograd version of https://github.com/jcjohnson/neural-style☆43Mar 20, 2016Updated 10 years ago
- PyTorch implementation of Deformable Convolution☆411Feb 17, 2019Updated 7 years ago
- 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.
- Support powerful visual logging in PyTorch.☆104Aug 18, 2017Updated 8 years ago
- auto-tuning momentum SGD optimizer☆287Mar 24, 2019Updated 7 years ago
- Fast Scattering Transform with CuPy/PyTorch☆295Feb 22, 2020Updated 6 years ago
- Notes for torch maintainers/modders☆10Mar 29, 2016Updated 9 years ago
- Totally Versatile Miscellanea for Pytorch☆477Mar 20, 2022Updated 4 years ago
- A language-agnostic interface to TensorBoard☆780Nov 9, 2017Updated 8 years ago
- a lightweight and simple logger for Machine Learning☆126Nov 9, 2020Updated 5 years ago