an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors - View it on GitHub