an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors
☆119May 26, 2025Updated 9 months ago
Alternatives and similar repositories for pytorch-extension
Users that are interested in pytorch-extension are comparing it to the libraries listed below
Sorting:
- CuPy fused PyTorch neural networks ops☆273Feb 15, 2018Updated 8 years ago
- ☆45Apr 25, 2017Updated 8 years ago
- Generalized Compressed Network Search with PyTorch☆26Nov 1, 2017Updated 8 years ago
- Deeper DCGAN with AE stabilization☆38Mar 20, 2024Updated last year
- PyTorch C++ Extension Example☆15Mar 4, 2018Updated 8 years ago
- Convolutional network training using Torch☆18Sep 18, 2016Updated 9 years ago
- Tutorial for building a custom CUDA function for Pytorch☆522Jan 25, 2019Updated 7 years ago
- DNI(Decoupled Neural Interfaces using Synthetic Gradients) implementation with Torch☆30Aug 30, 2016Updated 9 years ago
- Experiments with multiple classifiers to classify facial expressions☆14Feb 26, 2016Updated 10 years ago
- Pytorch implementation of "Forward Thinking: Building and Training Neural Networks One Layer at a Time"☆65Jun 14, 2017Updated 8 years ago
- VLFeat (partial) FFI wrapper for Torch7☆12Mar 23, 2016Updated 9 years ago
- ScatWave is a Torch implementation of scattering using CUDA☆21Jan 18, 2018Updated 8 years ago
- Generate captions for an image using convolutional and recurrent networks☆12Feb 25, 2016Updated 10 years ago
- A Thrift codec for Torch☆30Oct 21, 2021Updated 4 years ago
- Tools for PyTorch☆223Aug 10, 2022Updated 3 years ago
- LMDB for Torch☆27Sep 4, 2017Updated 8 years ago
- Unsupervised Learning of Multi-Frame Optical Flow with Occlusions☆112Nov 7, 2019Updated 6 years ago
- lua apply function for cutorch☆17Jan 5, 2017Updated 9 years ago
- Training Very Deep Neural Networks Without Skip-Connections☆589Jun 9, 2018Updated 7 years ago
- Examples of C extensions for PyTorch☆256Feb 12, 2023Updated 3 years ago
- ☆82Oct 19, 2017Updated 8 years ago
- random pytorch hacks☆26Jul 26, 2017Updated 8 years ago
- Online Lua REPL with Torch support.☆17Aug 28, 2015Updated 10 years ago
- PyTorch implementation of Deformable Convolution☆411Feb 17, 2019Updated 7 years ago
- Data structures, algorithms, and ML/NLP tools in Lua.☆26Aug 31, 2017Updated 8 years ago
- A neural assembly compiler for pyTorch based on adaptive-neural-compilation☆27Mar 27, 2018Updated 7 years ago
- Torch FFI-bindings for NNPACK☆31May 26, 2017Updated 8 years ago
- Photographic Image Synthesis with Cascaded Refinement Networks - Pytorch Implementation☆64Feb 14, 2018Updated 8 years ago
- Nervana Neon kernels in Torch☆19Nov 5, 2015Updated 10 years ago
- Support powerful visual logging in PyTorch.☆104Aug 18, 2017Updated 8 years ago
- autograd version of https://github.com/jcjohnson/neural-style☆43Mar 20, 2016Updated 9 years ago
- [adversarial] examples and training cost☆19Jun 29, 2016Updated 9 years ago
- Totally Versatile Miscellanea for Pytorch☆477Mar 20, 2022Updated 3 years ago
- Notes for torch maintainers/modders☆10Mar 29, 2016Updated 9 years ago
- ☆222Feb 15, 2020Updated 6 years ago
- auto-tuning momentum SGD optimizer☆288Mar 24, 2019Updated 6 years ago
- Low precision Torch nn library using uint8_t GEMM (experiment)☆19Mar 13, 2016Updated 9 years ago
- PyTorch wrapper for FFTs☆317Oct 28, 2018Updated 7 years ago
- A utility library around PyTorch☆244Dec 26, 2022Updated 3 years ago