vnmabus / dcor

Distance correlation and related E-statistics in Python
https://dcor.readthedocs.io
MIT License
145 stars 26 forks source link

Accelerate the rowwise AVL implementation of distance covariance using the GPU #17

Open vnmabus opened 4 years ago

vnmabus commented 4 years ago

Numba supports GPU programming, but most numpy functions are unsupported, which makes almost impossible sharing code for the GPU and CPU implementations.

If, however, any charitable soul wants to try to implement an alternative version of rowwise using the AVL implementation, accelerated using the GPU via Numba, it would be very helpful.