pytorch-extension
by sniklaus
an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors
AI summary
Hadamard product layer
A PyTorch CUDA extension that uses CuPy to compute the Hadamard product of two tensors.
- stars
- 117
- forks
- 14
- watching
- 6
Similar projects
Found by comparing what the projects do, not just their names.
C++ extension library
Enables the creation of custom C++ extensions with CUDA support in PyTorch
Tensor library
Provides a PyTorch interface for working with NumPy arrays and tensors.
CuPy ops
A Python implementation of PyTorch neural networks operations using CuPy.
Computer Vision Toolkit
A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision.
Message passing abstraction
Decoupled Neural Interfaces using Synthetic Gradients for PyTorch
GPU tensor library
A Python library providing tensors and dynamic neural networks with strong GPU acceleration
Memory optimizer for deep learning models
A lightweight wrapper around PyTorch to prevent CUDA out-of-memory errors and optimize model execution
FFT transformer
Provides an efficient wrapper around CUDA FFTs for PyTorch transformations
I3D weight transfer
Transfers pre-trained I3D network weights from TensorFlow to PyTorch
Deep learning tutorial
A hands-on introduction to deep learning using PyTorch, explaining mathematical concepts through code examples
Image translator
An implementation of an image-to-image translation algorithm using deep learning and PyTorch
Decoupled Neural Interface
An implementation of synthetic gradients to decouple neural network layers and enable scalable communication between them
Density estimator library
PyTorch implementations of algorithms for density estimation using invertible transformations.
Union Find Algorithm
An implementation of connected components labeling using a block-based union find algorithm
Brain MRI segmenter
A PyTorch implementation of the U-Net architecture for brain MRI segmentation using pre-trained weights and a specific deep learning algorithm
