sniklaus / pytorch-extension

an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors
118Updated last month

Alternatives and similar repositories for pytorch-extension:

Users that are interested in pytorch-extension are comparing it to the libraries listed below