Regarding gnn-cv and gpu support

[chlwjd1234]

Hi
I recently read the interesting paper "Descriptors-free Collective Variables From Geometric Graph Neural Networks", and wanna try this gnn cv by myself. ( The following git is referred from the paper , https://github.com/jintuzhang/gnncv/tree/master )
Is it possible to calculate this gnn-cv in plumed with gpu ?

Thanks in advance.