Grid extension + KAN with CUDA

[LouisD91]

Hello,

I tried using KANs on a GPU and apply grid extension. It raises multiple errors and I needed to modify the code for it to run. Performances are good but grid extension returned some NaNs. I had to perform the least square algorithm on the CPU to get rid of the NaNs and yet the performance significantly dropped after applying grid extension.