GPU support - Installation extras of CuPy and Implementation of Cuda kernels

transferwise / hisel

Feature selection tool based on Hilbert-Schmidt Independence Criterion

Apache License 2.0

2 stars 0 forks source link

Closed claudio-tw closed 1 year ago

claudio-tw commented 1 year ago

This PR implements the modifications to hisel that are required to support the computation of Gram matrix on GPU using CuPy