pytorch-extension
Hadamard product layer
A PyTorch CUDA extension that uses CuPy to compute the Hadamard product of two tensors.
an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors
117 stars
6 watching
14 forks
Language: Python
last commit: over 1 year ago cudacupydeep-learningpythonpytorch
Related projects:
| Repository | Description | Stars |
|---|---|---|
 pytorch/extension-cpp | Enables the creation of custom C++ extensions with CUDA support in PyTorch | 1,031 |
 wkentaro/pytorch-for-numpy-users | Provides a PyTorch interface for working with NumPy arrays and tensors. | 691 |
 szagoruyko/pyinn | A Python implementation of PyTorch neural networks operations using CuPy. | 274 |
 kaiyangzhou/dassl.pytorch | A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision. | 1,236 |
 koz4k/dni-pytorch | Decoupled Neural Interfaces using Synthetic Gradients for PyTorch | 236 |
| pytorch/pytorch | A Python library providing tensors and dynamic neural networks with strong GPU acceleration | 84,978 |
 rentruewang/koila | A lightweight wrapper around PyTorch to prevent CUDA out-of-memory errors and optimize model execution | 1,823 |
 locuslab/pytorch_fft | Provides an efficient wrapper around CUDA FFTs for PyTorch transformations | 315 |
 hassony2/kinetics_i3d_pytorch | Transfers pre-trained I3D network weights from TensorFlow to PyTorch | 532 |
 stared/thinking-in-tensors-writing-in-pytorch | A hands-on introduction to deep learning using PyTorch, explaining mathematical concepts through code examples | 331 |
 mrzhu-cool/pix2pix-pytorch | An implementation of an image-to-image translation algorithm using deep learning and PyTorch | 428 |
 andrewliao11/dni.pytorch | An implementation of synthetic gradients to decouple neural network layers and enable scalable communication between them | 119 |
 ikostrikov/pytorch-flows | PyTorch implementations of algorithms for density estimation using invertible transformations. | 577 |
 zsef123/connected_components_pytorch | An implementation of connected components labeling using a block-based union find algorithm | 109 |
 mateuszbuda/brain-segmentation-pytorch | A PyTorch implementation of the U-Net architecture for brain MRI segmentation using pre-trained weights and a specific deep learning algorithm | 730 |