feature request: drift correction computation on GPU

[WeissShahaf]

Dear Devs,
the drift estimation in its different forms takes a very long time.
Is it possible to move some of the computations for drift estimation and correction to the GPU? now that torch is in spikeinterface?

Thanks.
Shahaf

[alejoe91]

Hi @WeissShahaf

Have you tried the dredge_fast preset? That's the fastest implementation and it should take less than half of the recording duration.

It would be possible to implement some methods with GPU acceleration, but it will take months and a dedicated project...

[WeissShahaf]

i see. thanks!