Piece-wise rigid motion correction

Hi,

We currently use the piece-wise rigid motion correction from caiman.
The FIOLA article mentions the possibility to extend the motion correction for non-rigid motion.
Do you have plans to implement the piece-wise rigid motion correction using GPU?