Using numba for GPU acceleration ?

soft-matter / trackpy

Python particle tracking toolkit

http://soft-matter.github.io/trackpy

Other

441 stars 131 forks source link

Using numba for GPU acceleration ? #686

Open b-grimaud opened 2 years ago

b-grimaud commented 2 years ago

Hi,

I've seen that the idea of using GPUs for acceleration has been brought up before (#521, #304). As it has been a few years, I was wondering if there are any reasonable options today that could be leveraged to make use of GPUs without extensive rewriting ? I think numba has made a lot of progress regarding CUDA support.

Thanks !